Spark中的word2vec窗口大小

Question

Spark中的word2vec窗口大小

machine-learningapache-sparkword2vec

3

有没有办法改变在 pyspark.mllib.feature.Word2Vec 中使用的窗口大小？还是它被永久地固定为5？这似乎是一个非常重要的特性。

我在这里没有看到相关选项: https://spark.apache.org/docs/1.4.1/api/scala/index.html#org.apache.spark.mllib.feature.Word2Vec

- jamesmf

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Holden · Accepted Answer

没错，查看Word2Vec代码，我们可以看到它是一个私有值。如果你想的话，可以重写Word2Vec来改变窗口大小。我已经创建了一个JIRA（https://issues.apache.org/jira/browse/SPARK-10299），以允许设置窗口大小，因为1.5已经处于RC阶段，所以这可能要等到1.6才能实现。