Entropie vs indice de Gini
Dans cet exercice, vous allez comparer la justesse sur l'ensemble de test de dt_entropy à celle d'un autre arbre nommé dt_gini. L'arbre dt_gini a été entraîné sur le même jeu de données avec les mêmes paramètres, sauf pour le critère d'information, qui a été réglé sur l'indice de Gini à l'aide du mot-clé 'gini'.
X_test, y_test, dt_entropy, ainsi que accuracy_gini (qui correspond à la justesse sur l'ensemble de test obtenue par dt_gini) sont disponibles dans votre espace de travail.
Cette activité fait partie du cours
Machine Learning avec des modèles à base d'arbres en Python
Instructions de l’exercice
- Importez
accuracy_scoredepuissklearn.metrics. - Prédisez les étiquettes de l'ensemble de test à l'aide de
dt_entropyet assignez le résultat ày_pred. - Évaluez la justesse sur l'ensemble de test de
dt_entropyet assignez le résultat àaccuracy_entropy. - Examinez
accuracy_entropyetaccuracy_gini.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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}')