CoolFace
Modelpublic

sosoai/multilingual-mpnet-base-v2-embedding-all-safetensor

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
3likes52downloads
Model Card

paraphrase-multilingual-mpnet-base-v2-embedding-all

This model is a fine-tuned version of paraphrase-multilingual-mpnet-base-v2 on the following datasets: squad, newsqa, LLukas22/cqadupstack, LLukas22/fiqa, LLukas22/scidocs, deepset/germanquad, LLukas22/nq.

Usage (Sentence-Transformers)

Using this model becomes easy when you have sentence-transformers installed:

pip install -U sentence-transformers

Then you can use the model like this:

python
from sentence_transformers import SentenceTransformer
sentences = ["This is an example sentence", "Each sentence is converted"]

model = SentenceTransformer('LLukas22/paraphrase-multilingual-mpnet-base-v2-embedding-all')
embeddings = model.encode(sentences)
print(embeddings)

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 1E+00
  • —per device batch size: 40
  • —effective batch size: 120
  • —seed: 42
  • —optimizer: AdamW with betas (0.9,0.999) and eps 1E-08
  • —weight decay: 2E-02
  • —D-Adaptation: True
  • —Warmup: True
  • —number of epochs: 15
  • —mixedprecisiontraining: bf16

Training results

EpochTrain LossValidation Loss
00.0850.0625
10.05980.0554
20.04840.0518
30.04050.0485
40.03410.0463
50.02870.0454
60.02430.0445
70.02070.0426
80.01770.0424
90.01530.0421
100.01340.0417
110.0120.0411
120.0110.0414

Evaluation results

Epochtop_1top_3top_5top_10top_25
00.2610.3510.3840.4220.459
10.2720.3650.40.4390.477
20.2760.370.4040.4430.481
30.2920.3910.4260.4650.503
40.2950.3950.4310.470.51
50.2990.40.4370.4760.514
60.3060.4040.440.4780.515
70.3090.410.4450.4850.521
80.310.4110.4480.4870.524
90.3150.4170.4540.4930.529
100.3190.420.4570.4950.53
110.3230.4240.460.4970.531
120.3240.4270.4640.5010.536

Framework versions

  • —Transformers: 4.25.1
  • —PyTorch: 2.0.0.dev20230210+cu118
  • —PyTorch Lightning: 1.8.6
  • —Datasets: 2.7.1
  • —Tokenizers: 0.13.1
  • —Sentence Transformers: 2.2.2

Additional Information

This model was trained as part of my Master's Thesis 'Evaluation of transformer based language models for use in service information systems'. The source code is available on Github.