Vizualizarea importanței caracteristicilor
În acest exercițiu, vei determina care caracteristici au fost cele mai predictive conform regresorului de tip random forests rf pe care l-ai antrenat într-un exercițiu anterior.
În acest scop, vei desena un grafic cu bare orizontale al importanței caracteristicilor, așa cum a fost evaluată de rf. Din fericire, acest lucru se poate face cu ușurință datorită capabilităților de vizualizare ale pandas.
Am creat un obiect pandas.Series numit importances, care conține numele caracteristicilor ca index și importanțele lor ca valori. De asemenea, matplotlib.pyplot este disponibil ca plt, iar pandas ca pd.
Acest exercițiu face parte din cursul
Machine Learning cu modele bazate pe arbori în Python
Instrucțiuni pentru exercițiu
Apelează metoda
.sort_values()peimportancesși atribuie rezultatul variabileiimportances_sorted.Apelează metoda
.plot()peimportances_sortedși setează argumentele:kindla'barh'colorla'lightgreen'
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Create a pd.Series of features importances
importances = pd.Series(data=rf.feature_importances_,
index= X_train.columns)
# Sort importances
importances_sorted = ____
# Draw a horizontal barplot of importances_sorted
____.____(____='____', ____='____')
plt.title('Features Importances')
plt.show()