From ec866dbeb27867a92c243cfc39d4f836b9c40d86 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 21 Apr 2021 17:07:22 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py
index 62d6b575..a80d58b2 100644
--- a/modules/deeplearning/icing.py
+++ b/modules/deeplearning/icing.py
@@ -414,6 +414,9 @@ class IcingIntensityNN:
def build_evaluation(self):
self.train_accuracy = tf.keras.metrics.BinaryAccuracy(name='train_accuracy')
self.test_accuracy = tf.keras.metrics.BinaryAccuracy(name='test_accuracy')
+ self.test_auc = tf.keras.metrics.AUC(name='test_auc')
+ self.test_recall = tf.keras.metrics.Recall(name='test_recall')
+ self.test_precision = tf.keras.metrics.Precision(name='test_precision')
self.train_loss = tf.keras.metrics.Mean(name='train_loss')
self.test_loss = tf.keras.metrics.Mean(name='test_loss')
--
GitLab