Question

我正在使用Word2Vec的gensim实现。我有以下代码段：

print('training model')
model = Word2Vec(Sentences(start, end))
print('trained model:', model)
print('vocab:', model.vocab.keys())

当我在python2中运行它时，它按预期运行。最后的印刷品是词汇表中的所有单词。

但是，如果我在python3中运行它，我会收到错误：

trained model: Word2Vec(vocab=102, size=100, alpha=0.025)
Traceback (most recent call last):
  File "learn.py", line 58, in <module>
    train(to_datetime('-4h'), to_datetime('now'), 'model.out')
  File "learn.py", line 23, in train
    print('vocab:', model.vocab.keys())
AttributeError: 'Word2Vec' object has no attribute 'vocab'

发生了什么事？ gensim word2vec与python3不兼容吗？

Answer 1

您是否在这两个地方使用相同版本的gensim？ Gensim 1.0.0将vocab移动到辅助对象，因此在1.0.0之前的gensim版本中（在Python 2或3中），您可以使用：

model.vocab

...在gensim 1.0.0+中你应该使用（在Python 2或3中）......

model.wv.vocab

Gensim word2vec在python3缺少词汇

1 个答案: