使用XGBClassifier转储带有要素地图的XGBoost模型

Question

我想将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

Answer 1

我会回答我自己的问题。

在https://xgboost.readthedocs.io/en/latest/python/python_api.html中，首先使用get_booster()函数获取此模型的底层xgboost。助推器具有可用的dump_model功能。

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