CommencezCommencez gratuitement

Évaluer l'arbre de classification

Maintenant que vous avez entraîné votre premier arbre de classification, il est temps d'évaluer sa performance sur l'ensemble de test. Vous utiliserez pour cela la métrique de justesse, qui correspond à la proportion de prédictions correctes faites sur l'ensemble de test.

Le modèle entraîné dt de l'exercice précédent est déjà chargé dans votre espace de travail, ainsi que la matrice des caractéristiques de l'ensemble de test X_test et le tableau des étiquettes y_test.

Cette activité fait partie du cours

Machine Learning avec des modèles à base d'arbres en Python

Voir le cours

Instructions de l’exercice

  • Importez la fonction accuracy_score depuis sklearn.metrics.

  • Prédisez les étiquettes de l'ensemble de test et assignez le tableau obtenu à y_pred.

  • Évaluez la justesse sur l'ensemble de test de dt en appelant accuracy_score() et assignez la valeur à acc.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Import accuracy_score
from ____.____ import ____

# Predict test set labels
y_pred = ____.____(____)

# Compute test set accuracy  
acc = ____(____, ____)
print("Test set accuracy: {:.2f}".format(acc))
Modifier et exécuter le code