doc=["This is a sentence","This is another sentence"]
documents=[doc.strip().split(" ") for doc in doc1 ]
model = doc2vec.Doc2Vec(documents, size = 100, window = 300, min_count = 10, workers=4)
我得到 AttributeError: 'list' object has no attribute 'words' 因为 Doc2vec() 的输入文档不是正确的 LabeledSentence 格式。我希望下面的示例可以帮助您理解格式。
documents = LabeledSentence(words=[u'some', u'words', u'here'], labels=[u'SENT_1'])
更多细节在这里:http :
//rare-technologies.com/doc2vec-tutorial/ 但是,我通过使用 TaggedLineDocument() 从文件中获取输入数据解决了这个问题。
文件格式:一份文档 = 一行 = 一个 TaggedDocument 对象。单词应该已经被预处理并用空格分隔,标签是从文档行号自动构建的。
sentences=doc2vec.TaggedLineDocument(file_path)
model = doc2vec.Doc2Vec(sentences,size = 100, window = 300, min_count = 10, workers=4)
要获取文档向量:您可以使用 docvecs。更多详细信息:https ://radimrehurek.com/gensim/models/doc2vec.html#gensim.models.doc2vec.TaggedDocument
docvec = model.docvecs[99]
其中 99 是我们想要的向量的文档 ID。如果标签是整数格式(默认情况下,如果您使用 TaggedLineDocument() 加载),请像我一样直接使用整数 id。如果标签是字符串格式,请使用“SENT_99”。这类似于 Word2vec