语言: 德语
标签:

• 语义
• 句子转换器
• 句子相似度
  数据集:
• STS

德语STS模型

STS开发集（德语）

87.9%

STS测试集（德语）

84.3%

STS处理流程

!pip install -U sentence-transformers  
from sentence_transformers import SentenceTransformer  
model = SentenceTransformer('..模型路径..')  
sentences1 = ['猫坐在外面',  
             "一个男人在弹吉他",  
             '这部新电影太棒了']  

sentences2 = ['狗在花园里玩耍',  
              "一个女人在看电视",  
              '这部新电影非常精彩']  
embeddings1 = model.encode(sentences1, convert_to_tensor=True)  
embeddings2 = model.encode(sentences2, convert_to_tensor=True)  
cosine_scores = util.pytorch_cos_sim(embeddings1, embeddings2)  

for i in range(len(sentences1)):  
    for j in range(len(sentences2)):  
        print(cosine_scores[i][j]))  
"""  
猫坐在外面 狗在花园里玩耍  相似度: 0.1259  
猫坐在外面 一个女人在看电视 相似度: 0.0567  
猫坐在外面 这部新电影非常精彩 相似度: 0.0557  
一个男人在弹吉他 狗在花园里玩耍  相似度: 0.1031  
一个男人在弹吉他 一个女人在看电视 相似度: 0.0098  
一个男人在弹吉他 这部新电影非常精彩  相似度: 0.0828  
这部新电影太棒了 狗在花园里玩耍  相似度: 0.1008  
这部新电影太棒了 一个女人在看电视  相似度: 0.0674  
"""