下载扩展包,然后 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