Downloads · 30 days
3.2K
0% of all-time downloads
e5-base-sts-en-de
danielheinz/e5-base-sts-en-de
e5-base-sts-en-de is a feature extraction model from danielheinz. Use it when you need embeddings to search or compare text. It is set up for transformers. The card lists the license as mit.
INFO: The model is being continuously updated.
All-time downloads
3.1M
Public
Parameters
278M
2.2 GB on disk
Likes
17
Public
Repo makeup
Click a slice to open those files.
.safetensors1.1 GB · 98%
At a glance
- Task
- Feature Extraction
- Library
- transformers
- License
- mit
- Model type
- xlm-roberta
- Access
- Public
- Created
- Dec 21, 2023
- Updated
- Jan 14, 2024
- SHA
- a24243f5
Model card
From the Hugging Face model README
INFO: The model is being continuously updated.
The model is a multilingual-e5-base model fine-tuned with the task of semantic textual similarity in mind.
Model Training
The model has been fine-tuned on the German subsets of the following datasets:
The training procedure can be divided into two stages:
- training on paraphrase datasets with the Multiple Negatives Ranking Loss
- training on semantic textual similarity datasets using the Cosine Similarity Loss
Results
The model achieves the following results:
- 0.920 on stsb's validation subset
- 0.904 on stsb's test subset
- Task
- Feature Extraction
- Library
- transformers
- Type
- xlm-roberta
- License
- mit
- Languages
- de
- Hf Inference
- live
- Created
- Dec 21, 2023
- Updated
- Jan 14, 2024