From 14d7a1f50d71de242844b2312257ab5a281790d5 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 3 May 2021 18:03:07 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 8a5c92c5..26c797d6 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -474,9 +474,10 @@ 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)
+ if NumClasses == 2:
+ 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