From 9b594e1778038a965fd667c7063b57dbce29bc04 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 28 Oct 2021 15:00:26 -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 c8d9ff47..84fa53f9 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -914,15 +914,16 @@ class IcingIntensityNN:
ckpt.restore(ckpt_manager.latest_checkpoint)
- self.test_loss.reset_states()
- self.test_accuracy.reset_states()
+ self.reset_test_metrics()
for data0, data1, label in self.test_dataset:
ds = tf.data.Dataset.from_tensor_slices((data0, data1, label))
ds = ds.batch(BATCH_SIZE)
for mini_batch_test in ds:
self.predict(mini_batch_test)
- print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result())
+ f1, mcc = self.get_metrics()
+ print('loss, acc: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy(), self.test_recall.result().numpy(),
+ self.test_precision.result().numpy(), self.test_auc.result().numpy(), f1.numpy(), mcc.numpy())
labels = np.concatenate(self.test_labels)
self.test_labels = labels
--
GitLab