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。