Namen Bedeutung Grundstück verfügen nach Vorverarbeitung

Question

Bevor ein Modell baut Ich mag diese

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

und nach dem Bauprozess eines Feature Bedeutung Grundstück machen Skalierung

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

Das Problem besteht darin, dass statt Feature-Namen erhalten wir f0, f1, f2, f3 etc. ..... Wie man die Namen der Features zurückgibt?

dank

Answer 1

zuerst erhalten wir Liste mit Namen von Funktionen, bevor Vorverarbeitung

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

Dann

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

, die alle