efederici's picture
Create README.md
3bd4b49
|
raw
history blame
718 Bytes
metadata
pipeline_tag: text-classification
language:
  - it
tags:
  - cross-encoder
  - sentence-similarity
  - transformers

Cross-Encoder


Bridget Riley, COOL EDGE

Training Data

This model was trained on a custom ranking dataset.

Usage and Performance

from sentence_transformers import CrossEncoder
model = CrossEncoder('efederici/cross-encoder-distilbert-it')
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').