为什么Gensim doc2vec给出AttributeError:'list'对象没有属性'words'？

Question

我正在尝试通过使用以下代码来实验gensim doc2vec.据我从教程中理解,它应该工作.但是它给出了AttributeError:'list'对象没有属性'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的精彩教程，请也阅读一下。