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

cvičení

Použití entropie jako kritéria

V tomto cvičení natrénuješ klasifikační strom na datasetu Wisconsin Breast Cancer s použitím entropie jako informačního kritéria. Využiješ všech 30 příznaků v datasetu, který je rozdělený na 80 % trénovací a 20 % testovací část.

X_train i pole štítků y_train máš k dispozici ve svém pracovním prostoru.

Pokyny

100 XP

  • Importuj DecisionTreeClassifier z sklearn.tree.

  • Vytvoř instanci DecisionTreeClassifier s názvem dt_entropy s maximální hloubkou 8.

  • Nastav informační kritérium na 'entropy'.

  • Natrénuj dt_entropy na trénovací sadě.