可在线运行的notebook示例:在kaggle网站搜索 gte-large 查看
首先需要安装文本转向量的依赖包:
接着导入gte模型:
然后即可将文本转换为向量:
最后我们计算两个句子之间的余弦相似度,可以看到相似度非常接近1,可以认为两个句子相似: