ROC AUC
La curva ROC que trazaste en el último ejercicio parecía prometedora.
Ahora calcularás el área bajo la curva ROC, junto con las demás métricas de clasificación que hayas utilizado anteriormente.
Las funciones confusion_matrix
y classification_report
han sido precargadas para ti, junto con el modelo logreg
que construiste previamente, además de X_train
, X_test
, y_train
, y_test
. Además, las etiquetas del conjunto de pruebas predichas por el modelo se almacenan como y_pred
y las probabilidades de que las observaciones del conjunto de pruebas pertenezcan a la clase positiva se almacenan como y_pred_probs
.
También se ha creado un modelo knn
y se han impreso las métricas de rendimiento en la consola, para que puedas comparar los valores roc_auc_score
, confusion_matrix
y classification_report
entre los dos modelos.
Este ejercicio forma parte del curso
Aprendizaje supervisado con scikit-learn
Instrucciones de ejercicio
- Importa
roc_auc_score
. - Calcula e imprime la puntuación ROC AUC, pasando las etiquetas de prueba y las probabilidades de clase positivas predichas.
- Calcula e imprime la matriz de confusión.
- Llama a
classification_report()
.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Import roc_auc_score
____
# Calculate roc_auc_score
print(____(____, ____))
# Calculate the confusion matrix
print(____(____, ____))
# Calculate the classification report
print(____(____, ____))