我们可以在pyspark的ParamGridBuilder中使用for循环吗？

Question

下面的代码将参数添加到paramGridBuilder中，而不在pyspark中循环。在

from pyspark.ml.tuning import ParamGridBuilder
paramGrid = ParamGridBuilder()\
        .addGrid(lr.regParam, [0.1, 0.01]) \
        .addGrid(lr.fitIntercept, [False, True])\
        .addGrid(lr.elasticNetParam, [0.0, 0.5, 1.0])\
        .build()

我有一本这样的字典

^{pr2}$

我们可以用一个循环来构建ParamgridBuilder吗？它能工作吗？在

^{3}$

Answer 1

您可以使用reduce函数：

from functools import reduce
paramGrid  = reduce(
    lambda a,b: a.addGrid(*b),
    dict.items(),
    ParamGridBuilder(),
).build

或者用for循环

^{pr2}$