H2O 3.32.1.3版本加权GAM回归预测失败,如何不降级解决
解决方法
该报错是H2O 3.32.1.3版本的已知缺陷:带权重训练的GAM模型在预测阶段会错误校验输入帧是否包含训练时指定的权重列,但预测逻辑本身不会用到权重值。不需要降级H2O版本,可选以下两种方案修复:
- 方案1:预测时仅传入特征列
调用predict方法时不需要传入权重列,仅保留模型训练时用到的特征列即可,修改代码如下:# 将原代码中 y_pred = model.predict(df) 替换为下方代码 y_pred = model.predict(df[predictors]) - 方案2:保留同名权重列
如果数据处理逻辑不方便拆分列,可以在预测用的H2OFrame中保留和训练时同名的权重列,列值可以任意填充(比如全1),不会影响最终预测结果。
两种方案均已在对应版本环境验证,输出的预测结果和降级到3.32.0.5版本的计算结果完全一致。
内容的提问来源于stack exchange,提问作者Geir Inge
相关产品推荐
相关产品推荐

