PhilipMay/stsb_multi_mt
Viewer β’ Updated β’ 86.3k β’ 7.7k β’ 68
How to use danielheinz/e5-base-sts-en-de with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("feature-extraction", model="danielheinz/e5-base-sts-en-de") # Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("danielheinz/e5-base-sts-en-de")
model = AutoModel.from_pretrained("danielheinz/e5-base-sts-en-de")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.
The model has been fine-tuned on the German subsets of the following datasets:
The training procedure can be divided into two stages:
The model achieves the following results: