fcakyon / confplot Goto Github PK

Hello, I have used this package for my ml multiclass modeling alot. I like the formatting and the level of details in the plot. But recently I have noticed that the coloring of the plot are broken. I suspect that it might be related to recent updates in some of the dependencies.
Here is what I get with the following code

# import package
import confplot
from pandas import DataFrame
import numpy as np
# assume you have a confusion matrix array like this
array = 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]] )
# convert it to a pandas dataframe
df_cm = DataFrame(array, index=range(1, 7), columns=range(1, 7))
# create and save confusion matrix plot as "cm_plot.png"
confplot.plot_confusion_matrix_from_matrix(df_cm)

As you can see the diagonal and the last row/columns are not in the color they used to be. I dug into the code and it looks like the plot rendering neglects the facecolor updates in the quadmesh.