From 0720db0c79ca15b64909081c8ab657f0f599230a Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 12 Jul 2021 15:59:55 -0500
Subject: [PATCH] snapshot... changes to more easily switch between binary and
multiclass
---
modules/deeplearning/icing_cnn.py | 15 ++++++++-------
1 file changed, 8 insertions(+), 7 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 1cb21724..0fcde076 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -544,13 +544,14 @@ 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)
- self.test_true_neg(labels, pred)
- self.test_true_pos(labels, pred)
- self.test_false_neg(labels, pred)
- self.test_false_pos(labels, pred)
+ if NumClasses == 2:
+ self.test_auc(labels, pred)
+ self.test_recall(labels, pred)
+ self.test_precision(labels, pred)
+ self.test_true_neg(labels, pred)
+ self.test_true_pos(labels, pred)
+ self.test_false_neg(labels, pred)
+ self.test_false_pos(labels, pred)
def reset_test_metrics(self):
self.test_loss.reset_states()
--
GitLab