snapshot

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)