From 2f3421d508097e4efe6e2ae9cb8574017a8b20eb Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 28 Apr 2021 10:51:58 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 49ef4a33..15517dad 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -206,15 +206,15 @@ class IcingIntensityNN:
label = np.where(label == -1, 0, label)
# binary, two class
- #label = np.where(label != 0, 1, label)
- #label = label.reshape((label.shape[0], 1))
-
- keep = (label == 0) | (label == 3) | (label == 4) | (label == 5) | (label == 6)
- data = data[keep,]
- label = label[keep]
label = np.where(label != 0, 1, label)
label = label.reshape((label.shape[0], 1))
+ # keep = (label == 0) | (label == 3) | (label == 4) | (label == 5) | (label == 6)
+ # data = data[keep,]
+ # label = label[keep]
+ # label = np.where(label != 0, 1, label)
+ # label = label.reshape((label.shape[0], 1))
+
if CACHE_DATA_IN_MEM:
self.in_mem_data_cache[key] = (data, label)
--
GitLab