模型:

cross-encoder/stsb-TinyBERT-L-4

英文

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')。

您还可以只使用Transformers AutoModel类,无需使用sentence_transformers。