Visualizing Classification Results: Confusion Star and Confusion Gear

Amalia Luque, Mirko Mazzoleni, Alejandro Carrasco and Antonio Ferramosca

Notebook to visualize a Confusion Matrix

It contains the following 2 principal functions:

• PlotConfusionStar

Plot a confusion star (or a confusion gear)

• VideoConfusionStar

Generate a video file visualizing a sequence of confusion matrices in the form of confusion stars (or confusion gears)

To get examples of confusion matrices 2 additional functions are provided:

• GetConfusionMatrix

Reads an Excel file containing a confusion matrix

• GetSequenceConfusionMatrices

Reads an Excel file containing a sequence of confusion matrices

All the examples are based on simple classifiers applied to:

• The MNIST dataset
• A 10-class reduced version of the Abalone dataset
• The CIFAR-100 dataset

Required files

The following files are required:

• MNIST_cm.xlsx
• Abalone-10_cm.xlsx
• CIFAR-100_cm.xlsx
• MNIST_cm_sequence.xlsx

Installing instructions

To run the code, install the packages in the file "environment.yml" with the command

conda env create -f environment.yml

using the Anaconda python distribution