From 7364d8285fdb08652d76e5caf51874541a40c790 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 29 Apr 2021 12:25:33 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 613eadf3..8aea3c05 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -465,6 +465,12 @@ class IcingIntensityNN:
pred = self.model(inputs, training=False)
t_loss = self.loss(labels, pred)
+ 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 do_training(self, ckpt_dir=None):
if ckpt_dir is None:
--
GitLab