From ad4afeb87731c97f67647b6121e5d2a0b3586bee Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 29 Oct 2021 12:04:29 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 11 ++++++-----
1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 7fa5f375..b7bc8a75 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1085,8 +1085,9 @@ def run_evaluate_static_new(data_dct, num_lines, num_elems, ckpt_dir_s_path, fli
ckpt_dir_s = os.listdir(ckpt_dir_s_path)
ckpt_dir = ckpt_dir_s[0]
- probs_2d_s = []
- preds_2d_s = []
+ probs_2d_dct = {flvl: None for flvl in flight_levels}
+ preds_2d_dct = {flvl: None for flvl in flight_levels}
+
for flvl in flight_levels:
nn = IcingIntensityNN()
nn.flight_level = flvl
@@ -1109,10 +1110,10 @@ def run_evaluate_static_new(data_dct, num_lines, num_elems, ckpt_dir_s_path, fli
preds_2d = preds.reshape((num_lines, num_elems))
probs_2d = probs.reshape((num_lines, num_elems))
- probs_2d_s.append(probs_2d)
- preds_2d_s.append(preds_2d)
+ probs_2d_dct[flvl] = probs_2d
+ preds_2d_dct[flvl] = preds_2d
- return preds_2d_s, probs_2d_s
+ return preds_2d_dct, probs_2d_dct
if __name__ == "__main__":
--
GitLab