La curva ROC
Ahora que has construido un modelo de regresión logística para predecir el estado diabético, puedes trazar la curva ROC para visualizar cómo varían la tasa de verdaderos positivos y la tasa de falsos positivos a medida que cambia el umbral de decisión.
Las etiquetas de prueba, y_test
, y las probabilidades previstas de que las características de prueba pertenezcan a la clase positiva, y_pred_probs
, se han cargado previamente para ti, junto con matplotlib.pyplot
como plt
.
Crearás una curva ROC y luego interpretarás los resultados.
Este ejercicio forma parte del curso
Aprendizaje supervisado con scikit-learn
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# 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()