Een mozaïekplot van de confusion matrix tekenen
Hoewel het uitrekenen van de performancematrix leuk kan zijn, wordt het al snel repetitief als je meerdere confusion matrices van verschillende modellen nodig hebt. Gelukkig kan de methode .pred_table() de confusion matrix voor je berekenen.
Daarnaast kun je de output van de methode .pred_table() gebruiken om de confusion matrix te visualiseren met de functie mosaic().
churn en mdl_churn_vs_relationship zijn beschikbaar.
Deze oefening maakt deel uit van de cursus
Introductie tot regressie met statsmodels in Python
Oefeninstructies
- Importeer de functie
mosaic()uitstatsmodels.graphics.mosaicplot. - Maak
conf_matrixmet de methode.pred_table()en print deze. - Teken een mozaïekplot van de confusion matrix.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# 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()