word2vec获取词向量,查找相似词
首先安装gensim, 训练或者下载别人训练好的word2vec模型
# coding=utf-8
from gensim.models import Word2Vec
from gensim.models import KeyedVectors
#model = Word2Vec.load('./model/wiki.zh.text.model') #用模型
model = KeyedVectors.load_word2vec_format('./model/wiki.zh.text.vector') # 用转换成txt的词向量文件
testwords = ['孩子', '数学', '学术', '白痴', '篮球']
for i in range(5):
res = model.most_similar(testwords[i])
print(testwords[i])
print(res)
结果如下图