Tracer l'importance des fonctionnalités des noms après le prétraitement

Question

Avant de construire un modèle, je fais une mise à l'échelle de cette manière

X = StandardScaler(with_mean = 0, with_std = 1).fit_transform(X)

et après avoir construit un graphique d'importance des fonctionnalités

xgb.plot_importance(bst, color='red')
plt.title('importance', fontsize = 20)
plt.yticks(fontsize = 10)
plt.ylabel('fonctionnalités', fontsize = 20)

Le problème est que au lieu des noms des fonctionnalités, nous obtenons f0, f1, f2, f3 etc..... Comment revenir aux noms des fonctionnalités?

merci

Answer 1

Tout d'abord, nous obtenons la liste des noms des fonctionnalités avant le prétraitement

dtrain = xgb.DMatrix( X, label=y)
dtrain.feature_names

Ensuite

bst.get_fscore()
mapper = {'f{0}'.format(i): v for i, v in enumerate(dtrain.feature_names)}
mapped = {mapper[k]: v for k, v in bst.get_fscore().items()}
mapped
xgb.plot_importance(mapped, color='red')

c'est tout