Bert句嵌入模块sentence-transformers实战入门

参考链接：安装 — 句子转换器文档 (sbert.net)

1.安装

环境：python 3.6>=，pytorch 1.6.0>=，transformers v4.6.0>=。不支持 python 2.7

pytorch 安装参考 PyTorch安装报错_piukaty的博客-CSDN博客

transformers 安装：

pip install transformers

sentence-transformers安装：

pip install -U sentence-transformers

 检查：import sentence_transformers。能正常导入，无报错。在官方文档可能会让你输入一段英语小测验一下，其实也没必要。

2.加载中文预训练模型        

from sentence_transformers import SentenceTransformer

# 保存预训练模型的文件夹
save_path = "./bert_based_chinese"

# 加载模型
model = SentenceTransformer('distiluse-base-multilingual-cased-v1')
#