ParamGridBuilder

pyspark.ml.tuning。 ParamGridBuilder

构建器参数网格中使用网格搜索模型的选择。

例子

> > >pyspark.ml.classification进口LogisticRegression> > >lr=LogisticRegression()> > >输出=ParamGridBuilder()\baseOn({lrlabelCol:“l”})\baseOn([lrpredictionCol,“p”])\addGrid(lrregParam,(1.0,2.0])\addGrid(lr麦克斯特,(1,5])\构建()> > >预期=({lrregParam:1.0,lr麦克斯特:1,lrlabelCol:“l”,lrpredictionCol:“p”},{lrregParam:2.0,lr麦克斯特:1,lrlabelCol:“l”,lrpredictionCol:“p”},{lrregParam:1.0,lr麦克斯特:5,lrlabelCol:“l”,lrpredictionCol:“p”},{lrregParam:2.0,lr麦克斯特:5,lrlabelCol:“l”,lrpredictionCol:“p”})> > >len(输出)= =len(预期)真正的> > >所有([预期输出])真正的

方法

addGrid(参数值)

这个网格中给定的参数设置固定值。

baseOn(* args)

这个网格中给定的参数设置固定值。

构建()

构建并返回所有的组合参数指定的参数网格。

方法的文档

addGrid ( 参数:pyspark.ml.param.Param(任何],:列表(任何] )pyspark.ml.tuning.ParamGridBuilder

这个网格中给定的参数设置固定值。

参数必须是类的一个实例与实例相关的参数的参数(如估计或变压器)。

baseOn ( *arg游戏:联盟(ParamMap,元组(pyspark.ml.param.Param,任何]] )→ParamGridBuilder

这个网格中给定的参数设置固定值。接受一个参数字典或(参数值)对的列表。

构建 ( )→列表(ParamMap]

构建并返回所有的组合参数指定的参数网格。