CommencezCommencez gratuitement

Évaluer l'arbre optimal

Dans cet exercice, vous allez évaluer le score ROC AUC sur l'ensemble de test du meilleur modèle de grid_dt.

Pour y arriver, vous déterminerez d'abord la probabilité d'obtenir l'étiquette positive pour chaque observation de l'ensemble de test. Vous pouvez utiliser la méthode predict_proba() d'un classificateur scikit-learn pour calculer un tableau 2D contenant, par colonnes respectives, les probabilités des étiquettes de classe négative et positive.

Le jeu de données est déjà chargé et préparé pour vous (les variables numériques sont normalisées) ; il est séparé en 80 % entraînement et 20 % test. X_test, y_test sont disponibles dans votre espace de travail. De plus, nous avons aussi chargé l'objet GridSearchCV entraîné grid_dt que vous avez instancié dans l'exercice précédent. Notez que grid_dt a été entraîné comme suit :

grid_dt.fit(X_train, y_train)

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 roc_auc_score depuis sklearn.metrics.

  • Extrayez l'attribut .best_estimator_ de grid_dt et assignez-le à best_model.

  • Prédisez, pour l'ensemble de test, les probabilités d'obtenir la classe positive y_pred_proba.

  • Calculez le score ROC AUC sur l'ensemble de test test_roc_auc de best_model.

Exercice interactif pratique

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

# Import roc_auc_score from sklearn.metrics
____

# Extract the best estimator
best_model = ____

# Predict the test set probabilities of the positive class
y_pred_proba = ____

# Compute test_roc_auc
test_roc_auc = ____

# Print test_roc_auc
print('Test set ROC AUC score: {:.3f}'.format(test_roc_auc))
Modifier et exécuter le code