我正在构建一个用于逻辑回归的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(....)
如何使用训练集进行训练集验证来获取指标?
答案 0 :(得分:0)
请检查以下方法,应该适合您:
angular-cli.json