使用XGBClassifier转储带有特征图的XGBoost模型

时间:2019-09-28 19:58:37

标签: python machine-learning xgboost

我想将XGboost模型及其功能映射转储到文本文件中。可以这样做;

# https://xgboost.readthedocs.io/en/latest/python/python_intro.html
import xgboost as xgb
bst = xgb.train(param, dtrain, num_round, evallist)
# dump model with feature map
bst.dump_model('dump.raw.txt', 'featmap.txt')

但是,我正在使用XGBClassifier。方法dump_model在XGBClassifier中不可用。

from xgboost import XGBClassifier
xgboost_model = XGBClassifier()
xgboost_model.fit(x_train, y_train)
# line below can't work because dump_model is not available in XGBClassifier
xgboost_model.dump_model(‘dump.raw.txt’, 'featmap.txt’)  

如何使用XGBClassifier转储带有特征图的XGBoost模型?

我正在使用python 3.7

1 个答案:

答案 0 :(得分:0)

我会回答我自己的问题。

首先从https://xgboost.readthedocs.io/en/latest/python/python_api.html开始使用get_booster()函数来获得该模型的基础xgboost Booster。 Booster具有dump_model功能。

bst = xgboost_model.get_booster()
bst.dump_model('dump.raw.txt', 'featmap.txt')