La courbe ROC
Maintenant que vous avez construit un modèle de régression logistique pour prédire le statut du diabète, vous pouvez tracer la courbe ROC pour visualiser comment le taux de vrais positifs et le taux de faux positifs varient en fonction du seuil de décision.
Les étiquettes de test, y_test
, et les probabilités prédites des caractéristiques de test appartenant à la classe positive, y_pred_probs
, ont été préchargées pour vous, ainsi que matplotlib.pyplot
en tant que plt
.
Vous allez créer une courbe ROC, puis interpréter les résultats.
Cet exercice fait partie du cours
Apprentissage supervisé avec scikit-learn
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Import roc_curve
____
# Generate ROC curve values: fpr, tpr, thresholds
fpr, tpr, thresholds = ____(____, ____)
plt.plot([0, 1], [0, 1], 'k--')
# Plot tpr against fpr
plt.plot(____, ____)
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('ROC Curve for Diabetes Prediction')
plt.show()