【发布时间】:2015-05-10 19:04:33
【问题描述】:
Gensim 的official tutorial 明确指出可以继续训练(加载的)模型。我知道根据文档,无法继续训练从 word2vec 格式加载的模型。但即使从头生成模型然后尝试调用train 方法,也无法访问提供给train 的LabeledSentence 实例的新创建标签。
>>> sentences = [LabeledSentence(['first', 'sentence'], ['SENT_0']), LabeledSentence(['second', 'sentence'], ['SENT_1'])]
>>> model = Doc2Vec(sentences, min_count=1)
>>> print(model.vocab.keys())
dict_keys(['SENT_0', 'SENT_1', 'sentence', 'first', 'second'])
>>> sentence = LabeledSentence(['third', 'sentence'], ['SENT_2'])
>>> model.train([sentence])
>>> print(model.vocab.keys())
# At this point I would expect the key 'SENT_2' to be present in the vocabulary, but it isn't
dict_keys(['SENT_0', 'SENT_1', 'sentence', 'first', 'second'])
是否有可能继续在 Gensim 中使用新句子训练 Doc2Vec 模型?如果可以,如何实现?
【问题讨论】:
标签: neural-network gensim