- 多层感知机训练
- 功能介绍
- 参数说明
- 脚本示例
- 脚本代码
- 脚本运行结果
多层感知机训练
功能介绍
多层感知机多分类模型
参数说明
名称 | 中文名称 | 描述 | 类型 | 是否必须? | 默认值 | |
---|---|---|---|---|---|---|
layers | 神经网络层大小 | 神经网络层大小 | int[] | ✓ | ||
blockSize | 数据分块大小,默认值64 | 数据分块大小,默认值64 | Integer | 64 | ||
initialWeights | 初始权重值 | 初始权重值 | DenseVector | null | ||
vectorCol | 向量列名 | 向量列对应的列名,默认值是null | String | null | ||
featureCols | 特征列名数组 | 特征列名数组,默认全选 | String[] | null | ||
labelCol | 标签列名 | 输入表中的标签列名 | String | ✓ | ||
maxIter | 最大迭代步数 | 最大迭代步数,默认为 100 | Integer | 100 | ||
epsilon | 收敛阈值 | 迭代方法的终止判断阈值,默认值为 1.0e-6 | Double | 1.0E-6 | ||
l1 | L1 正则化系数 | L1 正则化系数,默认为0。 | Double | 0.0 | ||
l2 | 正则化系数 | L2 正则化系数,默认为0。 | Double | 0.0 |
脚本示例
脚本代码
mlpc = MultilayerPerceptronClassifier() \
.setVectorCol("bitmap") \
.setLabelCol("label") \
.setLayers([628, 100, 100]) \
.setMaxIter(100) \
.setPredictionCol("pred_label") \
.setPredictionDetailCol("pred_detail")
# mlpc.fit(batch_data.bo_mnist)
pipeline = Pipeline().add(mlpc)
model = pipeline.fit(batch_data.bo_mnist)
model.save('/tmp/mlpc.csv')
BatchOperator.execute()
脚本运行结果
-1,"{""schema"":[""model_id BIGINT,model_info VARCHAR,label_value BIGINT""],""param"":[""{\""vectorCol\"":\""\\\""bitmap\\\""\"",\""maxIter\"":\""100\"",\""layers\"":\""[628,100,100]\"",\""labelCol\"":\""\\\""label\\\""\"",\""predictionCol\"":\""\\\""pred_label\\\""\"",\""predictionDetailCol\"":\""\\\""pred_detail\\\""\""}""],""clazz"":[""com.alibaba.alink.pipeline.classification.MultilayerPerceptronClassificationModel""]}"
0,"0^{""vectorCol"":""\""bitmap\"""",""isVectorInput"":""true"",""layers"":""[628,100,100]"",""featureCols"":null}^"
......