IniziaInizia gratis

Usare l'entropia come criterio

In questo esercizio allenerai un albero di classificazione sul dataset Wisconsin Breast Cancer usando l'entropia come criterio informativo. Lo farai utilizzando tutte le 30 feature del dataset, che è suddiviso in 80% train e 20% test.

X_train e l'array di etichette y_train sono disponibili nel tuo workspace.

Questo esercizio fa parte del corso

Machine Learning con modelli ad alberi in Python

Visualizza il corso

Istruzioni dell'esercizio

  • Importa DecisionTreeClassifier da sklearn.tree.

  • Istanzia un DecisionTreeClassifier dt_entropy con profondità massima pari a 8.

  • Imposta il criterio informativo su 'entropy'.

  • Esegui il fit di dt_entropy sul training set.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Import DecisionTreeClassifier from sklearn.tree
from ____.____ import ____

# Instantiate dt_entropy, set 'entropy' as the information criterion
dt_entropy = ____(____=____, ____='____', random_state=1)

# Fit dt_entropy to the training set
____.____(____, ____)
Modifica ed esegui il codice