Fix matplotlib 3.3.x compatibility

sergioburdisso · Jul 17, 2020 · 6860284 · 6860284
1 parent 544ecb6
commit 6860284
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyss3/util.py b/pyss3/util.py
@@ -581,13 +581,13 @@ def __plot_confusion_matrices__(cms, classes, info='', max_colums=3, multilabel=
 
             if (axi % max_colums) == 0:
                 ax.set_ylabel('True', fontweight="bold")
-                ax.set_yticklabels(classes)
+                ax.set_yticklabels(classes[:cm.shape[0]])
             else:
                 ax.tick_params(labelleft=False)
 
             if axi + 1 > n_cms - max_colums or multilabel:
                 ax.set_xlabel('Predicted', fontweight="bold")
-                ax.set_xticklabels(classes)
+                ax.set_xticklabels(classes[:cm.shape[1]])
             else:
                 ax.tick_params(labelbottom=False)