From 45b1006467de913afe81ef31abadcfcabd48795c Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 10 Jul 2021 14:54:01 -0500
Subject: [PATCH] minor
---
modules/deeplearning/icing_cnn.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 11a559b5..be27c000 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -161,6 +161,7 @@ class IcingIntensityNN:
self.test_labels = []
self.test_preds = []
+ self.test_probs
self.learningRateSchedule = None
self.num_data_samples = None
@@ -717,6 +718,7 @@ class IcingIntensityNN:
labels = np.concatenate(self.test_labels)
preds = np.concatenate(self.test_preds)
+ self.test_probs = preds
preds = np.where(preds > 0.5, 1, 0)
self.test_labels = labels
--
GitLab