plot-confusion-matrix 0.0.2

A simple confusion matrix

Simple confusion matrix

Plot a simple confusion matrix using matplotlib

Usage:

confusion_matrix(matrix, labels)

labels is optional. Returns a matplotlib Figure

Example:

from plot_confusion_matrix import confusion_matrix
from matplotlib import pyplot

matrix = [[5, 2, 0],
		  [1, 2, 3],
		  [2, 1, 6]]

confusion_matrix(matrix)
pyplot.show(block=True)

Output:

confusion_matrix(matrix, ["A", "B", "C"])