Views
No views yet
SentenceTransformer(
(0): Transformer({'max_seq_length': 8192, 'do_lower_case': False}) with Transformer model: XLMRobertaModel
(1): Pooling({'word_embedding_dimension': 1024, 'pooling_mode_cls_token': True, 'pooling_mode_mean_tokens': False, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
(2): Normalize()
)| TASK | Snowflake | e1-EMB-German | e1-EMB-German vs. Snowflake |
|---|---|---|---|
| AmazonCounterfactualClassification | 0.6587 | 0.7152 | 5.65% |
| AmazonReviewsClassification | 0.3697 | 0.4577 | 8.80% |
| FalseFriendsGermanEnglish | 0.5360 | 0.5378 | 0.18% |
| GermanQuAD-Retrieval | 0.9423 | 0.9456 | 0.33% |
| GermanSTSBenchmark | 0.7499 | 0.8558 | 10.59% |
| MassiveIntentClassification | 0.6778 | 0.6826 | 0.48% |
| MassiveScenarioClassification | 0.7375 | 0.7494 | 1.19% |
| GermanDPR | 0.8367 | 0.8330 | -0.37% |
| MTOPDomainClassification | 0.9080 | 0.9259 | 1.79% |
| MTOPIntentClassification | 0.6675 | 0.7143 | 4.68% |
| PawsXPairClassification | 0.5887 | 0.5803 | -0.84% |
pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2
3# Download from the 🤗 Hub
4model = SentenceTransformer("embraceableAI/e1-EMB-German-Preview-v-0.1")
5# Run inference
6sentences = [
7 'The weather is lovely today.',
8 "It's so sunny outside!",
9 'He drove to the stadium.',
10]
11embeddings = model.encode(sentences)
12print(embeddings.shape)
13# [3, 1024]
14
15# Get the similarity scores for the embeddings
16similarities = model.similarity(embeddings, embeddings)
17print(similarities.shape)
18# [3, 3]@misc{bge-m3,
title={BGE M3-Embedding: Multi-Lingual, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation},
author={Jianlv Chen and Shitao Xiao and Peitian Zhang and Kun Luo and Defu Lian and Zheng Liu},
year={2024},
eprint={2402.03216},
archivePrefix={arXiv},
primaryClass={cs.CL}
}