Downloads · 30 days
8
2% of all-time downloads
distilroberta-base-stsb-cross-encoder
tomaarsen/distilroberta-base-stsb-cross-encoder
distilroberta-base-stsb-cross-encoder is a text classification model from tomaarsen. Use it when you need a label for a piece of text. It is set up for transformers.
This model was trained using SentenceTransformers Cross-Encoder class.
All-time downloads
461
Public
Parameters
82.1M
328 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors328 MB · 99%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- roberta
- Access
- Public
- Created
- Apr 17, 2024
- Updated
- Apr 17, 2024
- SHA
- e4fb3e96
Try a prompt
Model card
From the Hugging Face model README
Cross-Encoder
This model was trained using SentenceTransformers Cross-Encoder class.
Training Data
This model was trained on stsb. The model will predict a score between 0 and 1 for how semantically similarity two sentences are.
Usage and Performance
from sentence_transformers import CrossEncoder
model = CrossEncoder('tomaarsen/distilroberta-base-stsb-cross-encoder')
scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])
The model will predict scores for the pairs ('Sentence 1', 'Sentence 2') and ('Sentence 3', 'Sentence 4').
Model Card Author
I adapted this model card from https://huggingface.co/efederici/cross-encoder-bert-base-stsb by @efederici.
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- Created
- Apr 17, 2024
- Updated
- Apr 17, 2024