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
Instrucciones de ejercicio
Importa
DecisionTreeClassifier
desdesklearn.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
____.____(____, ____)