Gensim的Doc2vec中的Index2word引发了属性错误

时间:2017-03-31 17:01:18

标签: python gensim

我训练了一个doc2vec(gensim.models.Doc2Vec)模型,现在我正在使用这一行:

print(dict([(model.index2word[i], similarity) for i, similarity in enumerate(model.similar_by_word('igdumd32.dll@0x', topn=False))])['igdumd64.dll@0x'])

但它会产生此错误: AttributeError:'Doc2Vec'对象没有属性'index2word'

我正在使用gensim 1.0.1

你能帮忙吗?

1 个答案:

答案 0 :(得分:6)

在最近的gensim版本中,index2word字向量列表已移至模型的wv属性,因此您要说model.index2word,现在必须使用model.wv.index2word }。

(请注意,这仍然只是单词向量,只能通过" DM" dm=1 Doc2Vec模式进行训练。文档向量位于model.docvecs对象中,您可以在model.docvecs.offset2doctag中看到doc-vectors可能与之关联的字符串标记列表。)