snapshot...

modules/deeplearning/icing_fcn.py

@@ -973,9 +973,7 @@ class IcingIntensityFCN:
         pred_s = []
 
         for data in self.eval_dataset:
-            print(data[0].shape, data[1].shape)
             pred = self.model([data])
-            print(pred.shape, np.histogram(pred.numpy()))
             pred_s.append(pred)
 
         preds = np.concatenate(pred_s)
@@ -1122,13 +1120,14 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l
     for flvl in flight_levels:
         nn.flight_level = flvl
         nn.setup_eval_pipeline(data_dct, num_tiles)
-        nn.do_evaluate(ckpt_dir)
+        nn.do_evaluate(prob_thresh=prob_thresh)
 
         probs = nn.test_probs
-        if NumClasses == 2:
-            preds = np.where(probs > prob_thresh, 1, 0)
-        else:
-            preds = np.argmax(probs, axis=1)
+        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