diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 3b01ad2cb5d48d8b771a8abbaadd833b98d4f039..6207d2dee73a70680ef43517a892dd1f98f5ebc6 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -1103,7 +1103,7 @@ def run_evaluate_static_avg(data_dct, ll, cc, ckpt_dir_s_path, day_night='DAY', return ice_lons, ice_lats, preds_2d -def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', prob_thresh=0.5, +def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l1b_or_l2='both', prob_thresh=0.5, flight_levels=[0, 1, 2, 3, 4], use_flight_altitude=False): ckpt_dir_s = os.listdir(ckpt_dir_s_path) @@ -1116,7 +1116,7 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', p preds_dct = {flvl: None for flvl in flight_levels} for flvl in flight_levels: - nn = IcingIntensityNN(day_night=day_night, use_flight_altitude=use_flight_altitude) + nn = IcingIntensityNN(day_night=day_night, l1b_or_l2=l1b_or_l2, use_flight_altitude=use_flight_altitude) nn.flight_level = flvl nn.setup_eval_pipeline(data_dct, num_tiles) nn.build_model()