From b15d2fe27cd993880da2648c61a3d6deb3b523bc Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 7 Feb 2023 13:37:43 -0600
Subject: [PATCH] snapshot...
---
modules/util/plot_cm.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/modules/util/plot_cm.py b/modules/util/plot_cm.py
index eafcb4c6..34eabf66 100644
--- a/modules/util/plot_cm.py
+++ b/modules/util/plot_cm.py
@@ -11,6 +11,7 @@ import matplotlib.pyplot as plt
def plot_confusion_matrix_values(correct_labels, predict_labels, labels, title='Confusion matrix', filename='confusion_matrix', normalize=False):
cm = confusion_matrix(correct_labels, predict_labels)
+ return cm
def plot_confusion_matrix(cm, labels, title='Confusion matrix', filename = 'confusion_matrix', normalize=False):
--
GitLab