From 1998ce6cc3decbb0f5b5fa4c71469b5324bf23c7 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 13 Apr 2022 22:32:05 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index cceb93ad..32d00b88 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -401,8 +401,8 @@ class IcingIntensityFCN:
nda = np.zeros([nd_idxs.size])
nda[:] = self.flight_level
nda = tf.one_hot(nda, 5).numpy()
- nda = np.expand_dims(nda, axis=0)
- nda = np.expand_dims(nda, axis=0)
+ nda = np.expand_dims(nda, axis=1)
+ nda = np.expand_dims(nda, axis=1)
return data, nda
--
GitLab