pmml4s model.predict（）返回数组，而不是单个值

Question

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

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

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

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

有人知道为什么会这样吗？我输入预测值的方式是否错误，或者在序列化/反序列化中是否存在错误？

Answer 1

这是每个班级模型的确定性。在您的情况下，它表示概率为94.5％的4或概率为5.5％的5在简单的情况下，如果要接收值，则应选择索引作为最大值。

但是您可能会将此概率用于其他控制逻辑，例如在决策不明确时进行阈值处理（两个值的概率约为0.4，等等。