为什么Gensim doc2vec会给出AttributeError：＆＃39; list＆＃39;对象没有属性＆＃39; words＆＃39;？

Question

我正在尝试使用以下代码来试验gensim doc2vec。据我从教程中理解，它应该工作。但它会提供 AttributeError：＆＃39; list＆＃39;对象没有属性＆＃39;单词＆＃39;

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上阅读这个精彩的教程。