ComenzarEmpieza gratis

Utilizar la entropía como criterio

En este ejercicio, entrenarás un árbol de clasificación en el conjunto de datos Cáncer de Mama de Wisconsin utilizando la entropía como criterio de información. Lo harás utilizando las 30 características del conjunto de datos, que se divide en un 80 % de entrenamiento y un 20 % de prueba.

X_train así como el conjunto de etiquetas y_train están disponibles en tu espacio de trabajo.

Este ejercicio forma parte del curso

Machine learning con modelos basados en árboles en Python

Ver curso

Instrucciones de ejercicio

  • Importa DecisionTreeClassifier desde sklearn.tree.

  • Instancia un DecisionTreeClassifier dt_entropy con una profundidad máxima de 8.

  • Establece el criterio de información en 'entropy'.

  • Ajusta dt_entropy en el conjunto de entrenamiento.

Ejercicio interactivo práctico

Pruebe este ejercicio completando este código de muestra.

# 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
____.____(____, ____)
Editar y ejecutar código