从 Python 中的 H2OXGBoostEstimator 模型中提取原生 xgboost 模型

Question

是否可以从 Python 中的 H2OXGBoostEstimator 模型中提取原生 xgboost 模型 pickle 文件并通过原始 XGBoost Python API 读取？谢谢！

Answer 1

您可以尝试这两种“h2o-to-xgboost”方法从经过训练的 H2O 模型中提取 XGBoost 超参数和 DMatrix，这（根据文档）将为您提供完全相同的 XGBoost 原生 python 模型。

nativeXGBoostParam = h2oModelD.convert_H2OXGBoostParams_2_XGBoostParams()

nativeXGBoostInput = data.convert_H2OFrame_2_DMatrix(myX, y, h2oModelD)

        

nativeModel = xgb.train(dtrain=nativeXGBoostInput,

                        params=nativeXGBoostParam[0],                        

                        num_boost_round=nativeXGBoostParam[1])