From 36c2bbada865c32e2e67703f02baa85f4ce540ae Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 21 Apr 2021 16:28:07 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing.py | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py
index 31f3beca..a3a85cda 100644
--- a/modules/deeplearning/icing.py
+++ b/modules/deeplearning/icing.py
@@ -139,6 +139,9 @@ class IcingIntensityNN:
self.train_accuracy = None
self.test_loss = None
self.test_accuracy = None
+ self.test_auc = None
+ self.test_recall = None
+ self.test_precision = None
self.learningRateSchedule = None
self.num_data_samples = None
@@ -204,6 +207,12 @@ 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))
@@ -444,6 +453,9 @@ class IcingIntensityNN:
self.test_loss(t_loss)
self.test_accuracy(labels, pred)
+ self.test_auc(labels, pred)
+ self.test_recall(labels, pred)
+ self.test_precision(labels, pred)
def predict(self, mini_batch):
inputs = [mini_batch[0]]
--
GitLab