我想将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
答案 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')