模型:

cross-encoder/stsb-roberta-large

英文

Quora问题重复检测的交叉编码器

该模型使用 SentenceTransformers Cross-Encoder 类进行训练。

训练数据

该模型是在 STS benchmark dataset 上训练的。该模型将预测两个句子的语义相似度得分,得分介于0和1之间。

用法和性能

可以像这样使用预训练模型:

from sentence_transformers import CrossEncoder
model = CrossEncoder('model_name')
scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])

该模型将预测('句子1','句子2')和('句子3','句子4')这些句对的得分。

您还可以在不使用sentence_transformers的情况下,只使用Transformers AutoModel类来使用此模型。