XGBoost получает важность функции в виде списка столбцов вместо графика

Мне интересно, можем ли вы получить важность функции в виде списка столбцов вместо графика. Это то, что у меня есть

xg_reg = xgb.train(params=params, dtrain=data_dmatrix, num_boost_round=10)
import matplotlib.pyplot as plt

xgb.plot_importance(xg_reg)
plt.rcParams['figure.figsize'] = [5,5]
plt.show()

Что дает мне этот сюжет

введите здесь описание изображения

Вместо этого я хотел бы просто получить список основных функций, поскольку у меня более 800 различных функций.


person Snorrlaxxx    schedule 23.07.2020    source источник


Ответы (1)


Вы можете использовать xgb.get_score(). Вот несколько примеров: Как получить важность функции в xgboost?

person jared_mamrot    schedule 24.07.2020