从gensim中的word2vec中的数组中获取单词

时间:2015-10-10 22:35:24

标签: gensim word2vec

我刚开始使用http://rare-technologies.com/word2vec-tutorial/中提供的教程来试验word2vec格式gensim。如果我们需要原始输出向量,我们写:

model['computer']  

结果是:

array([-0.00449447, -0.00310097,  0.02421786, ...], dtype=float32)

如何获得有阵列的单词?所以,如果我写:

f=model['computer']

我怎样才能得到“计算机”这个词。用f?

1 个答案:

答案 0 :(得分:1)

我找到了此网站https://github.com/piskvorky/gensim/issues/381的解决方案:

word=model.most_similar(positive=[f],topn=1)
print(word[0][0])