pip install numpy
pip install reportlab
Quick drawing tool for the confusion matrix.
import numpy as np
from SIGraphicsKit import matrix
classes = ["ostrich", "swallow", "cuckoo", "chicken", "quail"]
data = np.array([
[50, 0, 0, 0, 0],
[1, 41, 5, 3, 0],
[2, 9, 34, 4, 1],
[1, 1, 8, 38, 2],
[1, 0, 2, 3, 44],
])
mx = matrix.Matrix(classes, data)
#Options
mx.cell_size = 10.0 #[mm]
mx.font_size = 14
mx.label_font_size = 7
mx.cell_color = "black" #black, red, yellow, green, blue, purple
mx.label_color = "black" #black, white
mx.line_type = "normal" #normal, dot
mx.percentage = False
mx.draw()