machine-learning - pmml4s model.predict() 返回数组而不是单个值

Question

我使用 sklearn2pmml 将我的决策树分类器序列化为 pmml 文件。我在java中使用pmml4s来反序列化模型并用它来预测。

我使用下面的代码对单个传入值进行预测。这应该返回 0/1/2/3/4/5/6。

Object[] result = model.predict(new String[]{"220"});

预测后的结果数组如下所示：

有谁知道为什么会这样？我输入预测值的方式是错误的还是序列化/反序列化有问题？

Answer 1

每个类别的模型都是确定性的。在您的情况下，这意味着它是 4，概率为 94.5% 或 5，概率为 5.5% 在简单的情况下，如果您想获得价值，您应该选择最大值的索引。

但是，您可以将此概率用于其他控制逻辑，例如在决策不明确时设置阈值（概率约为 0.4 的两个值等）