From 872e7ab014297b8537b6841288cdfa2b76c19b2a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 16 Nov 2021 12:07:47 -0600
Subject: [PATCH] snapshot...

---
 modules/deeplearning/icing_cnn.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 0638c9b2..0c04b7cd 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1005,7 +1005,7 @@ class IcingIntensityNN:
         self.do_evaluate(ckpt_dir)
 
 
-def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path):
+def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY'):
     ckpt_dir_s = os.listdir(ckpt_dir_s_path)
     cm_s = []
     prob_s = []
@@ -1015,7 +1015,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path):
         ckpt_dir = ckpt_dir_s_path + ckpt
         if not os.path.isdir(ckpt_dir):
             continue
-        nn = IcingIntensityNN()
+        nn = IcingIntensityNN(day_night=day_night)
         nn.run_restore(filename_l1b, filename_l2, ckpt_dir)
         cm_s.append(tf.math.confusion_matrix(nn.test_labels.flatten(), nn.test_preds.flatten()))
         prob_s.append(nn.test_probs.flatten())
-- 
GitLab