pytorch_bert_vec
pytorch_bert_vec copied to clipboard
nutalk
一条命令产生bert、albert句向量，用于相似度计算和文本分类等。
pytorch版Bert、Albert、Robert等模型的句向量生成
基于Transformers，albert_pytorch,用来简化生成句向量的代码，方便后续进行文本相似度计算等。
依赖
见requirements.txt
使用
效果
直接运行flask_api.py，用cosine距离判断句子的相似度，可以看到相似句子的bert和albert直接输出的向量相余弦距离比较小。
参考
句向量输出代码主要参考了Bert提取句子特征（pytorch_transformers）。
nutalk