From df3f5898aa7dff05ca2ed15f727c1aeefad84c49 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 3 May 2021 16:10:12 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 97ee728d..8a5c92c5 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -217,6 +217,7 @@ class IcingIntensityNN:
elif NumClasses == 3:
label = np.where(np.logical_or(label == 1, label == 2), 1, label)
label = np.where(np.invert(np.logical_or(label == 0, label == 1)), 2, label)
+ label = label.reshape((label.shape[0], 1))
if CACHE_DATA_IN_MEM:
self.in_mem_data_cache[key] = (data, label)
--
GitLab