Entropia versus índice de Gini
Neste exercício, você comparará a precisão do conjunto de teste de dt_entropy
com a precisão de outra árvore chamada dt_gini
. A árvore dt_gini
foi treinada no mesmo conjunto de dados usando os mesmos parâmetros, exceto pelo critério de informação que foi definido como o índice de gini usando a palavra-chave 'gini'
.
X_test
, y_test
, dt_entropy
, bem como accuracy_gini
, que corresponde à precisão do conjunto de teste obtida por dt_gini
, estão disponíveis no 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
accuracy_score
desklearn.metrics
. - Preveja os rótulos do conjunto de teste de
dt_entropy
e atribua o resultado ay_pred
. - Avalie a precisão do conjunto de teste de
dt_entropy
e atribua o resultado aaccuracy_entropy
. - Analise
accuracy_entropy
eaccuracy_gini
.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# Import accuracy_score from sklearn.metrics
from ____.____ import ____
# Use dt_entropy to predict test set labels
____= ____.____(____)
# Evaluate accuracy_entropy
accuracy_entropy = ____(____, ____)
# Print accuracy_entropy
print(f'Accuracy achieved by using entropy: {accuracy_entropy:.3f}')
# Print accuracy_gini
print(f'Accuracy achieved by using the gini index: {accuracy_gini:.3f}')