Rysowanie wykresu mozaikowego macierzy pomyłek
Ręczne obliczanie macierzy wydajności może być interesujące, ale szybko staje się żmudne, gdy trzeba porównywać wiele macierzy pomyłek dla różnych modeli. Na szczęście metoda .pred_table() zrobi to za ciebie automatycznie.
Dodatkowo możesz użyć wyniku zwróconego przez .pred_table() do wizualizacji macierzy pomyłek za pomocą funkcji mosaic().
Dostępne są: churn oraz mdl_churn_vs_relationship.
To ćwiczenie jest częścią kursu
Wprowadzenie do regresji z użyciem statsmodels w Pythonie
Instrukcje do ćwiczenia
- Zaimportuj funkcję
mosaic()zstatsmodels.graphics.mosaicplot. - Utwórz
conf_matrix, korzystając z metody.pred_table(), i wyświetl jej zawartość. - Narysuj wykres mozaikowy macierzy pomyłek.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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()