From b2f362f4f87173f91fe7c4ad8d32a83cf5187c4e Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 14 Apr 2022 10:20:24 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 11 ++---------
1 file changed, 2 insertions(+), 9 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index deb0f153..c8e175b1 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/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
--
GitLab