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

时间:2020-05-07 07:51:16

标签: machine-learning scikit-learn classification decision-tree pmml

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

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

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

预测后的结果数组如下:

enter image description here

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

0 个答案:

没有答案