XGBoost 将特征重要性作为列列表而不是绘图

Question

我想知道您是否可以将特征重要性作为列列表而不是绘图来获取。这就是我的

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 多种不同的功能。

Answer 1

您可以使用 xgb.get_score()。这里有一些例子：How to get feature importance in xgboost?

XGBoost 将特征重要性作为列列表而不是绘图

XGBoost get feature importance as a list of columns instead of plot

machine-learning

scikit-learn

xgboost