模型:
cross-encoder/stsb-roberta-large
该模型使用 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类来使用此模型。