Tracer un diagramme mosaïque de la matrice de confusion
Calculer la matrice de performance peut être amusant, mais cela devient fastidieux si vous avez besoin de plusieurs matrices de confusion pour différents modèles. Heureusement, la méthode .pred_table() peut calculer la matrice de confusion pour vous.
De plus, vous pouvez utiliser la sortie de la méthode .pred_table() pour visualiser la matrice de confusion avec la fonction mosaic().
churn et mdl_churn_vs_relationship sont disponibles.
Cette activité fait partie du cours
Introduction à la régression avec statsmodels en Python
Instructions de l’exercice
- Importez la fonction
mosaic()depuisstatsmodels.graphics.mosaicplot. - Créez
conf_matrixà l'aide de la méthode.pred_table()et affichez-la. - Tracez un diagramme mosaïque de la matrice de confusion.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# 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()