Zacznij terazZacznij za darmo

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

Zobacz kurs

Instrukcje do ćwiczenia

  • Wywołaj metodę .sort_values() na importances i przypisz wynik do importances_sorted.

  • Wywołaj metodę .plot() na importances_sorted i ustaw następujące argumenty:

    • kind na 'barh'
    • color na '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()
Edytuj i uruchom kod