1. Learn
    2. /
  2. Cursuri
    3. /
  3. Machine Learning cu modele bazate pe arbori în Python
Connected

exercițiu

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.

Instrucțiuni

100 XP

  • Apelează metoda .sort_values() pe importances și atribuie rezultatul variabilei importances_sorted.

  • Apelează metoda .plot() pe importances_sorted și setează argumentele:

    • kind la 'barh'
    • color la 'lightgreen'