Wizualizacja ważności cech
W tym ćwiczeniu sprawdzisz, które cechy miały największy wpływ na predykcje modelu – zgodnie z oceną regresora lasów losowych rf, który był trenowany w poprzednim ćwiczeniu.
W tym celu narysuj poziomy wykres słupkowy przedstawiający ważność cech wyznaczoną przez rf. Na szczęście możliwości wizualizacyjne biblioteki pandas sprawiają, że jest to proste.
Przygotowaliśmy obiekt pandas.Series o nazwie importances, który zawiera nazwy cech jako index oraz ich ważności jako wartości. Ponadto dostępny jest matplotlib.pyplot jako plt oraz pandas jako pd.
To ćwiczenie jest częścią kursu
Uczenie maszynowe z modelami drzewiastymi w Pythonie
Instrukcje do ćwiczenia
Wywołaj metodę
.sort_values()naimportancesi przypisz wynik doimportances_sorted.Wywołaj metodę
.plot()naimportances_sortedi ustaw następujące argumenty:kindna'barh'colorna'lightgreen'
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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()