1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Machine Learning with Tree-Based Models in Python
Connected

cvičení

Vizualizace důležitosti příznaků

V tomto cvičení zjistíš, které příznaky byly podle regresoru náhodného lesa rf – natrénovaného v předchozím cvičení – nejdůležitější pro predikci.

Sestrojíš horizontální sloupcový graf důležitosti příznaků, jak je vyhodnotil model rf. Díky vykreslovacím možnostem knihovny pandas to jde celkem snadno.

K dispozici máš objekt pandas.Series s názvem importances, kde jsou názvy příznaků jako index a jejich důležitosti jako hodnoty. Dále je dostupný matplotlib.pyplot jako plt a pandas jako pd.

Pokyny

100 XP

  • Zavolej metodu .sort_values() na objektu importances a výsledek ulož do proměnné importances_sorted.

  • Zavolej metodu .plot() na objektu importances_sorted a nastav následující argumenty:

    • kind na 'barh'
    • color na 'lightgreen'