为什么 Gensim doc2vec 给出 AttributeError: 'list' object has no attribute 'words'？

Question

我正在尝试使用以下代码来试验 gensim doc2vec。据我从教程中了解到，它应该可以工作。但是它给出了 AttributeError: 'list' object has no attribute 'words'。

from gensim.models.doc2vec import LabeledSentence, Doc2Vec
document = LabeledSentence(words=['some', 'words', 'here'], tags=['SENT_1']) 
model = Doc2Vec(document, size = 100, window = 300, min_count = 10, workers=4)

那么我做错了什么？请提供任何帮助。谢谢你。我正在使用 python 3.5 和 gensim 0.12.4

Answer 1

gensim.models.doc2vec 的输入应该是LabeledSentence 上的迭代器（比如一个列表对象）。试试：

model = Doc2Vec([document], size = 100, window = 1, min_count = 1, workers=1)

我减小了 window 和 min_count 的大小，以便它们对给定的输入有意义。如果您还没有，也可以在Doc2Vec 上阅读这个不错的教程。