Pretty Confusion Matrix for Python

The Pretty Confusion Matrix in Python with MATLAB like style, using seaborn and matplotlib.

This repository was forked and modified from Wagner’s Pretty print confusion matrix.

Example:

Installation

• Clone this repository:

   git clone https://github.com/phongsathorn1/pretty-confusion-matrix.git

• Install required packages:

   pip install -r requirements.txt

Usage

• Plot from numpy y_predict and y_actual vectors

   from pretty_cm import plot_from_data
   y_actual = np.array([1,2,3,4,5, 1,2,3,4,5, ...])
   y_predict = np.array([1,2,4,3,5, 1,2,3,4,4, ...])
   plot_from_data(y_actual, y_predict)

• Plot from numpy confusion matrix

   from pretty_cm import plot_from_confusion_matrix
   cm = np.array([[13,  0,  1,  0,  2,  0],
                  [ 0, 50,  2,  0, 10,  0],
                  [ 0, 13, 16,  0,  0,  3],
                  [ 0,  0,  0, 13,  1,  0],
                  [ 0, 40,  0,  1, 15,  0],
                  [ 0,  0,  0,  0,  0, 20]])
   plot_from_confusion_matrix(cm)

• Plot with custom labels

   plot_from_data(y_test, predic, 
                  columns=["Dog", "Cat", "Potato", "Car", "IU <3"])
   # -- or --
   plot_from_confusion_matrix(cm, columns=["Dog", "Cat", "Potato", "Car", "IU <3"])

  Result:

  

• Change figure size

   plot_from_data(y_test, predic, figsize=[6,6])
   # -- or --
   plot_from_confusion_matrix(cm, figsize=[6,6])

Licensing

The Pretty Confusion Matrix is licensed under Apache License, Version 2.0. see License for full license text.

References:

1. MATLAB confusion matrix

   • https://www.mathworks.com/help/nnet/ref/plotconfusion.html

   • https://www.mathworks.com/help/examples/nnet/win64/PlotConfusionMatrixUsingCategoricalLabelsExample_02.png