Trazar un diagrama mosaico de la matriz de confusión
Aunque calcular la matriz de rendimiento puede ser entretenido, se volvería tedioso si necesitas varias matrices de confusión de distintos modelos. Por suerte, el método .pred_table() puede calcular la matriz de confusión por ti.
Además, puedes usar la salida del método .pred_table() para visualizar la matriz de confusión con la función mosaic().
churn y mdl_churn_vs_relationship están disponibles.
Este ejercicio forma parte del curso
Introducción a la regresión con statsmodels en Python
Instrucciones del ejercicio
- Importa la función
mosaic()destatsmodels.graphics.mosaicplot. - Crea
conf_matrixusando el método.pred_table()e imprímelo. - Dibuja un diagrama mosaico de la matriz de confusión.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Import mosaic from statsmodels.graphics.mosaicplot
____
# Calculate the confusion matrix conf_matrix
conf_matrix = ____
# Print it
print(conf_matrix)
# Draw a mosaic plot of conf_matrix
____
plt.show()