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

cvičení

Entropie vs. Giniho index

V tomto cvičení porovnáš přesnost modelu dt_entropy na testovací sadě s přesností jiného stromu dt_gini. Strom dt_gini byl natrénován na stejném datasetu se stejnými parametry, až na informační kritérium, které bylo nastaveno na Giniho index pomocí klíčového slova 'gini'.

V pracovním prostředí máš k dispozici X_test, y_test, dt_entropy a také accuracy_gini, což je přesnost modelu dt_gini na testovací sadě.

Pokyny

100 XP
  • Importuj accuracy_score z sklearn.metrics.
  • Předpověz štítky testovací sady pomocí dt_entropy a výsledek ulož do y_pred.
  • Vyhodnoť přesnost modelu dt_entropy na testovací sadě a výsledek ulož do accuracy_entropy.
  • Prohlédni si hodnoty accuracy_entropy a accuracy_gini.