Skip to content
Snippets Groups Projects
Commit ff156558 authored by tomrink's avatar tomrink
Browse files

minor...

parent 5d7d71f1
Branches
No related tags found
No related merge requests found
...@@ -1010,7 +1010,7 @@ class IcingIntensityFCN: ...@@ -1010,7 +1010,7 @@ class IcingIntensityFCN:
self.do_evaluate(ckpt_dir) self.do_evaluate(ckpt_dir)
def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY', use_flight_altitude=False): def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY', l1b_or_l2='both', use_flight_altitude=False):
ckpt_dir_s = os.listdir(ckpt_dir_s_path) ckpt_dir_s = os.listdir(ckpt_dir_s_path)
cm_s = [] cm_s = []
prob_s = [] prob_s = []
...@@ -1020,7 +1020,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA ...@@ -1020,7 +1020,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA
ckpt_dir = ckpt_dir_s_path + ckpt ckpt_dir = ckpt_dir_s_path + ckpt
if not os.path.isdir(ckpt_dir): if not os.path.isdir(ckpt_dir):
continue continue
nn = IcingIntensityFCN(day_night=day_night, use_flight_altitude=use_flight_altitude) nn = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_or_l2, use_flight_altitude=use_flight_altitude)
nn.run_restore(filename_l1b, filename_l2, ckpt_dir) nn.run_restore(filename_l1b, filename_l2, ckpt_dir)
cm_s.append(tf.math.confusion_matrix(nn.test_labels.flatten(), nn.test_preds.flatten())) cm_s.append(tf.math.confusion_matrix(nn.test_labels.flatten(), nn.test_preds.flatten()))
prob_s.append(nn.test_probs.flatten()) prob_s.append(nn.test_probs.flatten())
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment