Heatmap for Confusion Matrix in Python | by Michał Marcińczuk, Ph.D. | Sep, 2024

One picture may be price 1000’s of phrases.

A confusion matrix is a handy approach to current the varieties of errors a machine studying mode makes. It’s an N by N grid with numbers, the place the worth within the [n, m] cell represents the variety of examples annotated with the n-th class acknowledged because the m-th class. On this tutorial, I’ll concentrate on making a confusion matrix and a heatmap. The colour palette will probably be used to show the sizes of various teams, making it straightforward to note similarities or important variations in group sizes. This type of visualization is useful once you take care of quite a few classes.

Here’s a visible rationalization of the weather of the confusion matrix.