如何在 PySpark Word2vec 模型中设置迭代次数?

How to set the number of iterations in PySpark Word2vec model?

我的代码:

 word2Vec = Word2Vec(vectorSize = 100, minCount = 100, inputCol = 'token', 
 outputCol = 'word2vec', seed=123)

我尝试了关键字 numIterations,
但无效

来自 Pyspark 文档:

setNumIterations(int numIterations)

设置迭代次数(默认:1),应小于或等于分区数。

感谢您的帮助。

参数似乎是maxIter: 试一试。

  final val maxIter: IntParam = new IntParam(this, "maxIter", 
     "maximum number of iterations (>= 0)", ParamValidators.gtEq(0))