From b972f2a5392530e04c86c9d70d16c633a847fdf5 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 14 May 2021 13:35:36 -0500
Subject: [PATCH] minor...
---
modules/deeplearning/icing_cnn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 5409e32c..39b7f588 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -463,14 +463,14 @@ class IcingIntensityNN:
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_f1 = tfa.metrics.F1Score(name='test_f1')
+ self.test_f1 = tfa.metrics.F1Score(NumClasses, name='test_f1')
self.test_recall = tf.keras.metrics.Recall(name='test_recall')
self.test_precision = tf.keras.metrics.Precision(name='test_precision')
else:
self.train_accuracy = tf.keras.metrics.SparseCategoricalAccuracy(name='train_accuracy')
self.test_accuracy = tf.keras.metrics.SparseCategoricalAccuracy(name='test_accuracy')
self.test_auc = tf.keras.metrics.AUC(name='test_auc')
- self.test_f1 = tfa.metrics.F1Score(name='f1_score')
+ self.test_f1 = tfa.metrics.F1Score(NumClasses, name='f1_score')
self.test_recall = tf.keras.metrics.Recall(name='test_recall')
self.test_precision = tf.keras.metrics.Precision(name='test_precision')
--
GitLab