在skip-gram和CBOW模型之间进行选择,以便在gensim中训练word2Vec

时间:2016-09-17 21:54:26

标签: nlp gensim word2vec

在培训 Word2Vec 模型时,是否可以在 Gensim 中选择Skip-gramCBOW模型?

感谢。

1 个答案:

答案 0 :(得分:10)

是。初始化参数sg控制模式。如果是True-ish(sg=1),则使用skip-gram;如果为False-ish(sg=0),则使用CBOW。

docs for gensim's Word2Vec class涵盖了此内容。