我已经训练并保存了H2O AutoML模型。重新加载后,当我使用预测方法时,出现以下错误: java.lang.IllegalArgumentException:测试/验证数据集具有一个非分类列“ response”,该列在训练数据中是分类的
在创建模型时我没有指定任何编码,但是现在出现此错误。谁能帮助我解决这个问题。
我们将不胜感激任何帮助。
答案 0 :(得分:0)
此问题与训练集中不存在的特定列中的新示例数据有关。 在这种情况下,我使用将列类型解析为数字(或字符串)。
def _convert_h2oframe_to_numeric(h2o_frame, training_columns):
for column in training_columns:
h2o_frame[column] = h2o_frame[column].asnumeric()
return h2o_frame
记得在训练和预测过程中使用这个函数。