ComeçarComece gratuitamente

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

Ver Curso

Instruções de exercício

  • Importe DecisionTreeClassifier de sklearn.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
____.____(____, ____)
Editar e executar código