Gensim:如何使用之前的word2vec模型重新训练doc2vec模型

时间:2016-08-31 14:21:27

标签: python gensim word2vec doc2vec

通过Doc2Vec建模,我训练了一个模型并保存了以下文件:

$TITAN_HOME

但是,我有一种新方法来标记文档并希望再次训练模型。因为已经从先前版本获得了单词向量。有没有办法重用该模型(例如,将之前的w2v结果作为训练的初始向量)?谁知道怎么做?

1 个答案:

答案 0 :(得分:1)

我已经发现,我们可以加载模型并继续训练。

model = Doc2Vec.load("old_model")
model.train(sentences)