From 51a20716181c4fe7a3013dd0d860c24cd7c5f16d Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 21 Dec 2021 09:59:07 -0600
Subject: [PATCH] snapshot
---
modules/util/util.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/modules/util/util.py b/modules/util/util.py
index e049667e..8b29f733 100644
--- a/modules/util/util.py
+++ b/modules/util/util.py
@@ -642,11 +642,9 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon
icing_prob_ds.dims[0].label = dim_0_name
icing_prob_ds.dims[1].label = dim_1_name
- pred_s = np.stack(pred_s, axis=-1)
- sum_pred = np.sum(pred_s, axis=2)
- sum_pred = np.where(sum_pred < 0, -1, sum_pred)
+ max_lvl = np.argmax(prob_s, axis=2)
- icing_pred_ds = h5f_out.create_dataset('sum_icing_column', data=sum_pred, dtype='i2')
+ icing_pred_ds = h5f_out.create_dataset('max_icing_probability_level', data=max_lvl, dtype='i2')
icing_pred_ds.attrs.create('coordinates', data='y x')
icing_pred_ds.attrs.create('grid_mapping', data='Projection')
icing_pred_ds.attrs.create('missing', data=-1)
--
GitLab