您的位置:首页 > 其它

对word2vec训练的model的结果操作笔记

2017-02-27 17:30 323 查看

存储、加载模型的方法如下:
>>> model.save('/tmp/mymodel')
>>> new_model = gensim.models.Word2Vec.load('/tmp/mymodel')
[/code]
使用模型Word2vec支持数种单词相似度任务:
model.most_similar(positive=['woman', 'king'], negative=['man'], topn=1)
[('queen', 0.50882536)]
model.doesnt_match("breakfast cereal dinner lunch".split())
'cereal'
model.similarity('woman', 'man')
.73723527
1
2
3
4
5
6
1
2
3
4
5
6
[/code]可以通过以下方式来得到单词的向量:
model['computer']  # raw NumPy vector of a word
array([-0.00449447, -0.00310097,  0.02421786, ...], dtype=float32)
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  word2vec