Skip to main content
 首页 » 编程设计

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

2024年12月31日30mayingbao

我的代码:

 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))