From a23b96087c9e2ef09391d69b3c4ed35a4e60a2f5 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 28 Apr 2021 10:19:36 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_cnn.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 7b209117..49ef4a33 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -535,14 +535,19 @@ class IcingIntensityNN:
self.test_loss.reset_states()
self.test_accuracy.reset_states()
+ self.test_auc.reset_states()
+ self.test_recall.reset_states()
+ self.test_precision.reset_states()
+
for data0, label in self.test_dataset:
ds = tf.data.Dataset.from_tensor_slices((data0, label))
ds = ds.batch(BATCH_SIZE)
for mini_batch in ds:
self.test_step(mini_batch)
- print('loss, acc : ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy())
- print('---------------------------------------------------------')
+ print('loss, acc, auc, recall, precision: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy(),
+ self.test_auc.result().numpy(), self.test_recall.result().numpy(), self.test_precision.result().numpy())
+ print('--------------------------------------------------')
ckpt_manager.save()
if self.DISK_CACHE and epoch == 0:
--
GitLab