From 725c2b8971c5cd7f839a0713f73034c4840d81b9 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 14 Dec 2021 10:34:39 -0600
Subject: [PATCH] switch to total column prediction for now - need more R&D
---
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 6b1e1342..a664f450 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1050,7 +1050,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA
return labels, prob_avg, cm_avg
-def run_evaluate_static(data_dct, ll, cc, ckpt_dir_s_path, flight_level=4, prob_thresh=0.5, satellite='GOES16', domain='FD'):
+def run_evaluate_static(data_dct, ll, cc, ckpt_dir_s_path, day_night='DAY', flight_level=4, prob_thresh=0.5, satellite='GOES16', domain='FD'):
num_elems = len(cc)
num_lines = len(ll)
cc = np.array(cc)
@@ -1065,7 +1065,7 @@ def run_evaluate_static(data_dct, ll, cc, ckpt_dir_s_path, flight_level=4, prob_
ckpt_dir = ckpt_dir_s_path + ckpt
if not os.path.isdir(ckpt_dir):
continue
- nn = IcingIntensityNN()
+ nn = IcingIntensityNN(day_night=day_night)
nn.flight_level = flight_level
nn.setup_eval_pipeline(data_dct, num_lines * num_elems)
nn.build_model()
--
GitLab