diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index acc119b684a32c63028ff66307bfa1584eb10aa8..ea211d6afe44571fbb0013e68d87889e9625a332 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1159,6 +1159,7 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_levels=[0, 1, 2, 3, 4], use_flight_altitude=False, use_max_cth_level=False): + flight_levels = flight_levels.copy() if not use_flight_altitude: flight_levels = [0] @@ -1187,8 +1188,6 @@ def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_le probs_dct[flvl] = model.test_probs.flatten() preds_dct[flvl] = model.test_preds.flatten() - - return preds_dct, probs_dct