首页 > 解决方案 > H2O AutoML 错误测试/验证数据集有一个非分类列,它在训练数据中是分类的“预测

问题描述

我已经训练并保存了我的 H2O AutoML 模型。重新加载后,当我使用 predict 方法时,出现以下错误: java.lang.IllegalArgumentException: Test/Validation dataset has a non-categorical column 'response' 这在训练数据中是分类的

创建模型时我没有指定任何编码,但我现在收到此错误。任何人都可以在这个问题上帮助我。

任何帮助将不胜感激。

标签: h2opredictautoml

解决方案


这个问题与训练集中不存在的特定列中的新示例数据有关。在这种情况下,我使用将列类型解析为数字(或字符串)。

def _convert_h2oframe_to_numeric(h2o_frame, training_columns):
    for column in training_columns:
        h2o_frame[column] = h2o_frame[column].asnumeric()
    return h2o_frame

请记住将此功能用于训练和预测过程。


推荐阅读