CommencerCommencer gratuitement

Dessiner une mosaïque de la matrice de confusion

Si le calcul de la matrice de performance peut être amusant, il deviendrait fastidieux si vous aviez besoin de plusieurs matrices de confusion de différents modèles. Heureusement, la méthode .pred_table() peut calculer la matrice de confusion pour vous.

En outre, vous pouvez utiliser la sortie de la méthode .pred_table() pour visualiser la matrice de confusion, à l'aide de la fonction mosaic().

churn et mdl_churn_vs_relationship sont disponibles.

Cet exercice fait partie du cours

Introduction à la régression avec statsmodels en Python

Afficher le cours

Instructions

  • Importez la fonction mosaic() à partir de statsmodels.graphics.mosaicplot.
  • Créez conf_matrix à l'aide de la méthode .pred_table() et imprimez-le.
  • Tracez un graphique en mosaïque de la matrice de confusion.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# 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()
Modifier et exécuter le code