From de4083cc41daaec9e2258c0ab9df4ef7607b96cb Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 8 Mar 2023 11:22:13 -0600 Subject: [PATCH] snapshot... --- modules/util/plot_cm.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/util/plot_cm.py b/modules/util/plot_cm.py index 64cfebe3..7e5c2a1d 100644 --- a/modules/util/plot_cm.py +++ b/modules/util/plot_cm.py @@ -32,14 +32,16 @@ def plot_confusion_matrix(cm, labels, title='Confusion matrix', filename = 'conf ''' if normalize: - cm = cm.astype('float')*10 / cm.sum(axis=1)[:, np.newaxis] + cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis] + cm *= 100 cm = np.nan_to_num(cm, copy=True) cm = cm.astype('int') np.set_printoptions(precision=2) fig = plt.figure(figsize=(3, 3), dpi=320, facecolor='w', edgecolor='k') ax = fig.add_subplot(1, 1, 1) - im = ax.imshow(cm, cmap='Oranges') + # im = ax.imshow(cm, cmap='Oranges') + im = ax.imshow(cm, cmap='Blues') classes = [re.sub(r'([a-z](?=[A-Z])|[A-Z](?=[A-Z][a-z]))', r'\1 ', x) for x in labels] classes = ['\n'.join(wrap(l, 40)) for l in classes] -- GitLab