从 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])

更多信息：

• 文档
• 例