From 755a52038e2e7611d211310306160e87b43ee3bd Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 13 Jul 2021 12:11:19 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/icing_cnn.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 5d601b54..a65f86d1 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -723,11 +723,16 @@ class IcingIntensityNN:
print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result())
labels = np.concatenate(self.test_labels)
+ self.test_labels = labels
+
preds = np.concatenate(self.test_preds)
self.test_probs = preds
- preds = np.where(preds > 0.5, 1, 0)
- self.test_labels = labels
+ if NumClasses == 2:
+ preds = np.where(preds > 0.5, 1, 0)
+ else:
+ preds = np.argmax(preds, axis=1)
+
self.test_preds = preds
self.h5f_tst.close()
--
GitLab