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

One picture may be value hundreds of phrases.

Picture by writer

A confusion matrix is a handy method 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 deal with making a confusion matrix and a heatmap. The colour palette shall be used to show the sizes of various teams, making it simple to note similarities or vital variations in group sizes. This sort of visualization is useful whenever you cope with quite a few classes.

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