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 do exercício
- Importe
accuracy_scoredesklearn.metrics. - Preveja os rótulos do conjunto de teste de
dt_entropye atribua o resultado ay_pred. - Avalie a precisão do conjunto de teste de
dt_entropye atribua o resultado aaccuracy_entropy. - Analise
accuracy_entropyeaccuracy_gini.
Exercício interativo prático
Experimente este exercício completando 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}')