(三)利用Wikipedia中文语料训练词向量word2vec——word2vec模型测试

通过前两篇文章,我们得到了word2vec模型:
(一)利用Wikipedia中文语料训练词向量word2vec——获取Wikipedia简体中文语料库
(二)利用Wikipedia中文语料训练词向量word2vec——分词、训练word2vec
下面,我们利用word2vec模型进行简单的使用和测试。

# 导入包
from gensim.models import Word2Vec

# 加载模型
wiki_word2vec_model = Word2Vec.load('word2vec_300.model')

# 计算相词语的相似度
testwords = ['苹果', '数学', '奖励', '资助', '创新']

for i in range(5):
    res = wiki_word2vec_model.most_similar(testwords[i])
    print(testwords[i])
    print(res)
    print('*'*20)

输出结果:
(三)利用Wikipedia中文语料训练词向量word2vec——word2vec模型测试_第1张图片

你可能感兴趣的:(MachineLearning,word2vec,gensim)