From e5174d8c9e2b8811503b95dac9dabc06402b3dab Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 15 May 2021 17:59:14 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/icing_cnn.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 648c8ab6..f77ec0e6 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -617,8 +617,12 @@ class IcingIntensityNN:
for mini_batch in ds:
self.test_step(mini_batch)
- print('loss, acc, auc, recall, precision: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy(),
- self.test_auc.result().numpy(), self.test_f1.result().numpy(), self.test_recall.result().numpy(), self.test_precision.result().numpy())
+ recall = self.test_recall.result().numpy()
+ precsn = self.test_precision.result().numpy()
+ f1 = 2 * (precsn * recall) / (precsn + recall)
+
+ print('loss, acc, auc, recall, precision, f1: ', 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(), f1)
print('--------------------------------------------------')
ckpt_manager.save()
--
GitLab