From d0165f67c0a9896d0ec9fac5d33291b989623c20 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Tue, 22 Mar 2022 17:17:08 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/icing_fcn.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 3d31fad0..05e2284d 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -400,6 +400,8 @@ class IcingIntensityFCN: nda = np.zeros([1]) nda[0] = self.flight_level nda = tf.one_hot(nda, 5).numpy() + nda = np.expand_dims(nda, axis=0) + nda = np.expand_dims(nda, axis=0) return data, nda -- GitLab