我训练了一个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
你能帮忙吗?
答案 0 :(得分:6)
在最近的gensim版本中,index2word
字向量列表已移至模型的wv
属性,因此您要说model.index2word
,现在必须使用model.wv.index2word
}。
(请注意,这仍然只是单词向量,只能通过" DM" dm=1
Doc2Vec模式进行训练。文档向量位于model.docvecs
对象中,您可以在model.docvecs.offset2doctag
中看到doc-vectors可能与之关联的字符串标记列表。)