Entropie par rapport à l'indice de Gini
Dans cet exercice, vous comparerez la précision de l'ensemble de test de dt_entropy à celle d'un autre arbre nommé dt_gini. dt_gini L'arbre de décision a été formé sur le même ensemble de données en utilisant les mêmes paramètres, à l'exception du critère d'information qui a été défini sur l'indice de Gini à l'aide du mot-clé « 'gini' ».
X_test, y_test, dt_entropy, ainsi que accuracy_gini, qui correspond à la précision du jeu de test obtenue par dt_gini, sont disponibles dans votre espace de travail.
Cet exercice fait partie du cours
Machine learning avec des modèles arborescents en Python
Instructions
- Importez
accuracy_scoreà partir desklearn.metrics. - Veuillez prédire les étiquettes de l'ensemble de test de
dt_entropyet attribuer le résultat ày_pred. - Veuillez évaluer la précision de l'ensemble de test de
dt_entropyet attribuer le résultat àaccuracy_entropy. - Révisions sur
accuracy_entropyetaccuracy_gini
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# 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}')