From 63a9e447d432a7aecb2284cfa7778105cd262a7a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 14 Apr 2022 09:44:33 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 13 ++++++-------
1 file changed, 6 insertions(+), 7 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 950c9715..deb0f153 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/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
--
GitLab