如何计算训练模型的对数损失?

时间:2016-05-03 17:30:24

标签: apache-spark apache-spark-mllib apache-spark-ml

我正在构建一个用于逻辑回归的ML管道。

val lr = new LogisticRegression()
lr.setMaxIter(100).setRegParam(0.001)

val pipeline = new Pipeline().setStages(Array(geoDimEncoder,clientTypeEncoder,
               devTypeDimIdEncoder,pubClientIdEncoder,tmpltIdEncoder,
               hourEncoder,assembler,lr))

val model = pipeline.fit(trainingDF)

现在,当训练模型时,我希望看到训练集的概率并计算某些验证参数,如对数丢失。但是,我无法使用“模型”找到它。

我到处都能找到唯一的东西是

model.transform(testDF).select(....)

如何使用训练集进行训练集验证来获取指标?

1 个答案:

答案 0 :(得分:0)

请检查以下方法,应该适合您:

angular-cli.json