pmml4s model.predict() returns 数组而不是单个值
pmml4s model.predict() returns array instead of single value
我使用 sklearn2pmml 将我的决策树分类器序列化为 pmml 文件。
我在 java 中使用 pmml4s 反序列化模型并用它来预测。
我使用下面的代码对单个传入值进行预测。这应该 return 0/1/2/3/4/5/6.
Object[] result = model.predict(new String[]{"220"});
预测后的结果数组如下所示：
有人知道为什么会这样吗？是我输入预测值的方式有误，还是 serialization/deserialization 有问题？
每个class都是模型的确定性。在您的情况下，这意味着它是 4，概率为 94.5% 或 5，概率为 5.5%
在简单的情况下，如果你想接收值，你应该选择最大值的索引。
但是，您可以将此概率用于其他控制逻辑，例如决策不明确时的阈值处理（概率为 ~0.4 的两个值，等等）
我使用 sklearn2pmml 将我的决策树分类器序列化为 pmml 文件。 我在 java 中使用 pmml4s 反序列化模型并用它来预测。
我使用下面的代码对单个传入值进行预测。这应该 return 0/1/2/3/4/5/6.
Object[] result = model.predict(new String[]{"220"});
预测后的结果数组如下所示：
有人知道为什么会这样吗？是我输入预测值的方式有误，还是 serialization/deserialization 有问题？
每个class都是模型的确定性。在您的情况下，这意味着它是 4，概率为 94.5% 或 5，概率为 5.5% 在简单的情况下，如果你想接收值，你应该选择最大值的索引。
但是，您可以将此概率用于其他控制逻辑，例如决策不明确时的阈值处理（概率为 ~0.4 的两个值，等等）