Gensim的word2vec中的层大小

时间:2019-02-06 17:44:36

标签: python python-3.x nlp gensim word2vec

当我开始训练word2vec模型时,会收到警告

  

考虑将图层大小设置为4的倍数以获得更高的性能

这听起来很简洁,但是我找不到在the documentation中对layer参数或类似内容的任何引用。

那么如何增加图层大小,如何确定合适的值?

1 个答案:

答案 0 :(得分:2)

图层大小仅表示可以通过size参数设置的单词向量的大小(尺寸)。默认值为100,但是您可以尝试使用128(例如4)的倍数。最佳大小取决于您的训练数据,必须根据经验确定。通常,更多数据意味着您可以选择更大的容量。