snapshot...

modules/deeplearning/icing_fcn.py

@@ -1122,15 +1122,8 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l
         nn.setup_eval_pipeline(data_dct, num_tiles)
         nn.do_evaluate(prob_thresh=prob_thresh)
 
-        probs = nn.test_probs
-        preds = nn.test_preds
-        # if NumClasses == 2:
-        #     preds = np.where(probs > prob_thresh, 1, 0)
-        # else:
-        #     preds = np.argmax(probs, axis=1)
-
-        probs_dct[flvl] = probs
-        preds_dct[flvl] = preds
+        probs_dct[flvl] = nn.test_probs.flatten()
+        preds_dct[flvl] = nn.test_preds.flatten()
 
     return preds_dct, probs_dct