Downloads · 30 days
40
1% of all-time downloads
efederici/cross-encoder-umberto-stsb
cross-encoder-umberto-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.
Downloads · 30 days
40
1% of all-time downloads
All-time downloads
3K
Public
Repo size
886 MB
Likes
0
Public
Click a slice to open those files.
.bin443 MB · 99%
From the Hugging Face model README
This model was trained using SentenceTransformers Cross-Encoder class.
<p align="center"> <img src="https://user-images.githubusercontent.com/7140210/72913702-d55a8480-3d3d-11ea-99fc-f2ef29af4e72.jpg" width="700"> </br> Marco Lodola, Monument to Umberto Eco, Alessandria 2019 </p>This model was trained on stsb. The model will predict a score between 0 and 1 how for the semantic similarity of two sentences.
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').