在gensim中加载Word2Vec模型时出错

时间:2015-08-19 17:17:41

标签: python gensim word2vec

我在加载word2vec存储库中提供的gensim模型时获得IList

AttributeError

这是一个已知问题吗?

3 个答案:

答案 0 :(得分:11)

解决了以下问题:

from gensim import models
w = models.Word2Vec.load_word2vec_format('GoogleNews-vectors-negative300.bin', binary=True)
print w["queen"]

答案 1 :(得分:3)

为了在不同的训练算法(Word2Vec,Fastext,WordRank,VarEmbed)之间共享单词向量查询代码,作者将单词向量的存储和查询分离为单独的KeyedVectors类。

已弃用word2vec类中的两个方法和多个属性。

方法

  • load_word2vec_format
  • save_word2vec_format

属性

  • syn0norm
  • syn0
  • 翻译
  • index2word

这些已移至KeyedVectors类。

升级到此版本后,您可能会获得有关已弃用方法或缺少属性的例外情况。

要删除例外,您应该使用

KeyedVectors.load_word2vec_format (instead ofWord2Vec.load_word2vec_format)
word2vec_model.wv.save_word2vec_format (instead of  word2vec_model.save_word2vec_format)
model.wv.syn0norm instead of  (model.syn0norm)
model.wv.syn0 instead of  (model.syn0)
model.wv.vocab instead of (model.vocab)
model.wv.index2word instead of (model.index2word)

答案 2 :(得分:0)

当前,由于models.Word2Vec已过时,您需要使用models.KeyedVectors.load_word2vec_format而不是models.Word2Vec.load_word2vec_format,如下所示。

from gensim import models
w = models.KeyedVectors.load_word2vec_format('model.bin', binary=True)