Downloads · 30 days
16
1% of all-time downloads
cross-encoder-bert-base-stsb
efederici/cross-encoder-bert-base-stsb
cross-encoder-bert-base-stsb is a text classification model from efederici. 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
3K
Public
Repo size
886 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin443 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Apr 4, 2022
- Updated
- Apr 4, 2022
- SHA
- 083a417f
Try a prompt
Datasets
Model card
From the Hugging Face model README
Cross-Encoder
This model was trained using SentenceTransformers Cross-Encoder class.
<p align="center"> <img src="https://upload.wikimedia.org/wikipedia/commons/f/f6/Edouard_Vuillard%2C_1920c_-_Sunlit_Interior.jpg" width="400"> </br> Edouard Vuillard, Sunlit Interior </p>Training Data
This model was trained on stsb. The model will predict a score between 0 and 1 how for the semantic similarity of two sentences.
Usage and Performance
from sentence_transformers import CrossEncoder
model = CrossEncoder('efederici/cross-encoder-umberto-stsb')
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').
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- it
- Created
- Apr 4, 2022
- Updated
- Apr 4, 2022