' Doc2Vec'对象没有属性' wv'

时间:2017-04-22 00:55:22

标签: python nlp gensim word2vec doc2vec

当我从pkl文件加载doc2vec模型时,我收到此错误。

    ---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-45-5ed9dc71f3a6> in <module>()
----> 1 temp_docky.infer_vector(['paypal'])

C:\Users\Laxmikant\Anaconda\lib\site-packages\gensim\models\doc2vec.pyc in infer_vector(self, doc_words, alpha, min_alpha, steps)
    750                 train_document_dm(self, doc_words, doctag_indexes, alpha, work, neu1,
    751                                   learn_words=False, learn_hidden=False,
--> 752                                   doctag_vectors=doctag_vectors, doctag_locks=doctag_locks)
    753             alpha = ((alpha - min_alpha) / (steps - i)) + min_alpha
    754 

C:\Users\Laxmikant\Anaconda\lib\site-packages\gensim\models\doc2vec_inner.pyx in gensim.models.doc2vec_inner.train_document_dm (./gensim/models/doc2vec_inner.c:5165)()
    406     # default vectors, locks from syn0/doctag_syn0
    407     if word_vectors is None:
--> 408        word_vectors = model.wv.syn0
    409     _word_vectors = <REAL_t *>(np.PyArray_DATA(word_vectors))
    410     if doctag_vectors is None:

AttributeError: 'Doc2Vec' object has no attribute 'wv'

你能帮忙解决这个错误吗?

1 个答案:

答案 0 :(得分:0)

该模型保存了哪个版本的gensim,以及您将其加载到哪个版本? (在最近的版本中,字矢量存储在模型中的方式已经改变。)

一般来说,gensim已经尝试确保旧模型正确加载,但是有些版本在这种向后兼容性方面存在疏忽......所以也值得尝试非常新的gensim,看看它是否加载你的旧模型没有这个错误。 (如果它出现同样的错误,请将其作为project discussion listGithub project bug tracker上的错误报告。)