Aan de slagGa gratis aan de slag

De ROC-curve

Nu je een logistiek regressiemodel hebt gebouwd om diabetes te voorspellen, kun je de ROC-curve plotten om te visualiseren hoe de true positive rate en false positive rate veranderen als de drempelwaarde verschuift.

De testlabels, y_test, en de voorspelde waarschijnlijkheden dat de testfeatures tot de positieve klasse behoren, y_pred_probs, zijn alvast voor je ingeladen, samen met matplotlib.pyplot als plt.

Je maakt een ROC-curve en interpreteert daarna de resultaten.

Deze oefening maakt deel uit van de cursus

Supervised Learning met scikit-learn

Cursus bekijken

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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()
Code bewerken en uitvoeren