Usando a entropia como um critério
Neste exercício, você treinará uma árvore de classificação no Conjunto de Dados de Câncer de Mama de Wisconsin usando a entropia como critério de informação. Você fará isso usando todos os 30 recursos do conjunto de dados, que é dividido em 80% de treinamento e 20% de teste.
X_train
bem como o conjunto de rótulos y_train
estão disponíveis em seu espaço de trabalho.
Este exercício faz parte do curso
Aprendizado de máquina com modelos baseados em árvores em Python
Instruções de exercício
Importe
DecisionTreeClassifier
desklearn.tree
.Instancie um
DecisionTreeClassifier
dt_entropy
com uma profundidade máxima de 8.Defina o critério de informação como
'entropy'
.Ajuste
dt_entropy
no conjunto de treinamento.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# 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
____.____(____, ____)