自然语言处理实战(一):king-man+woman

点点圈 提交于 2020-02-27 00:55:55

第一个word2vec例子:woman+king-man=queen。

1、加载二进制模型

直接使用Google开源模型GoogleNews-vectors-negative300.bin,使用gensim库来加载这个二进制模型:

from gensim import models
w=models.KeyedVectors.load_word2vec_format('GoogleNews-vectors-negative300.bin',binary=True)
2、打印结果
print('King-man+woman:')
print('')
print(w.wv.most_similar(positive=['woman','king'],negative=['man']))

输出:
在这里插入图片描述

3、woman和man的相似度
print('Similarity between man and woman:')
print(w.similarity('woman','man'))

输出:
在这里插入图片描述

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!