diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py index 62d6b5754d364b481f56fbed40f8f9bfcdd39240..a80d58b2a806df79689002df96c3d12e3d3367d3 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')