Создать дамп модели XGBoost с картой объектов с помощью XGBClassifier

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’)  

Как создать дамп модели XGBoost с картой объектов с использованием XGBClassifier?

Я использую 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')