diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 65c51921b73a0b08dac958522b394a84ae5a1dca..feb90a73c15e5b9257f45fe0a2d6b1babd9147aa 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1090,7 +1090,7 @@ class IcingIntensityFCN: def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY', l1b_or_l2='both', - satellite='GOES16', use_flight_altitude=False, out_file=None): + use_flight_altitude=False, out_file=None): ckpt_dir_s = os.listdir(ckpt_dir_s_path) cm_s = [] prob_s = [] @@ -1100,7 +1100,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 = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_or_l2, satellite=satellite, 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) cm_s.append(tf.math.confusion_matrix(nn.test_labels.flatten(), nn.test_preds.flatten())) prob_s.append(nn.test_probs.flatten())