license: apache-2.0 datasets:

• sentence-transformers/stsb language:
• en pipeline_tag: text-ranking library_name: sentence-transformers tags:
• transformers

语义文本相似度交叉编码器

本模型使用SentenceTransformers的交叉编码器类进行训练。

训练数据

本模型基于STS基准数据集训练，可预测两个句子之间0到1分值的语义相似度。

使用方式与性能表现

预训练模型可通过如下方式调用：

from sentence_transformers import CrossEncoder

model = CrossEncoder('cross-encoder/stsb-TinyBERT-L4')
scores = model.predict([('句子1', '句子2'), ('句子3', '句子4')])

模型将为文本对('句子1', '句子2')和('句子3', '句子4')预测相似度分值。

该模型也可脱离sentence_transformers库，直接使用Transformers的AutoModel类进行调用。