Dibujar un gráfico de mosaico de la matriz de confusión
Aunque calcular la matriz de rendimiento puede ser divertido, se volvería tedioso si se necesitaran múltiples matrices de confusión de diferentes modelos. Por suerte, el método .pred_table() puede calcular la matriz de confusión por usted.
Además, puede utilizar la salida del método .pred_table() para visualizar la matriz de confusión, utilizando 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 modelos estadísticos en Python
Instrucciones de ejercicio
- Importe la función
mosaic()destatsmodels.graphics.mosaicplot. - Cree
conf_matrixutilizando el método.pred_table()e imprímalo. - Dibuja un gráfico de mosaico de la matriz de confusión.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este 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()