From 84c2eed2c51a386aaf43b591621d1e778ed5182c Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 10 Jan 2022 16:42:02 -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 6d971d2c..426c07af 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -1022,7 +1022,7 @@ class IcingIntensityNN: self.do_evaluate(ckpt_dir) -def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY'): +def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY', use_flight_altitude=False): ckpt_dir_s = os.listdir(ckpt_dir_s_path) cm_s = [] prob_s = [] @@ -1032,7 +1032,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA ckpt_dir = ckpt_dir_s_path + ckpt if not os.path.isdir(ckpt_dir): continue - nn = IcingIntensityNN(day_night=day_night) + nn = IcingIntensityNN(day_night=day_night, use_flight_altitude=use_flight_altitude) 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