按照 mahout 中的甜甜圈分类示例,我构建了如下模型:
构建模型:
./mahout trainlogistic --input donut.csv --output ./model --target color --categories 2 --predictors x y a b c --types numeric --features 20 --passes 100 --rate 10
我这样评估模型:
./mahout runlogistic --input donut.csv --model model --auc --confusion
输出是:
AUC = 0.97
confusion: [[27.0, 13.0], [0.0, 0.0]]
entropy: [[-0.4, -0.3], [-1.2, -0.7]]
第一个命令在本地磁盘上生成了一个模型文件。如何使用此模型对新数据进行分类?有这个命令吗?还是我需要编写 Java 代码来加载该模型并对其进行分类?