word2vec获取词向量，查找相似词

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)

结果如下图