yunxier / pytorch_bert_vec

一条命令产生bert、albert句向量，用于相似度计算和文本分类等。

pytorch版Bert、Albert、Robert等模型的句向量生成

基于Transformers，albert_pytorch,用来简化生成句向量的代码，方便后续进行文本相似度计算等。

依赖

见requirements.txt

使用

1. 下载中文bert、albert的预训练模型。
2. 使用flask_api.py。

效果

直接运行flask_api.py，用cosine距离判断句子的相似度，可以看到相似句子的bert和albert直接输出的向量相余弦距离比较小。

参考

句向量输出代码主要参考了Bert提取句子特征（pytorch_transformers）。