下载扩展包,然后 pip install 下载的文件
词移距离计算代码示例
from text2vec import Similarity
sim = Similarity(similarity_type='wmd')
s1 = "工业硫酸"
s2 = ["硫酸",'硫酸钴','硫酸镍','硫酸铁','硫酸锌']
similar_list = []
for s in s2:
similar_list.append(sim.get_score(s1, s))
print(similar_list)
https://www.lfd.uci.edu/~gohlke/pythonlibs/#pyemd