diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index a8542c771aeefc349df44132407122a22f4c2693..648c8ab6cd35bc2a64f6ebd593e3d2b4ec319de5 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -512,7 +512,7 @@ class IcingIntensityNN: self.test_accuracy(labels, pred) if NumClasses == 2: self.test_auc(labels, pred) - self.test_f1(labels, pred) + #self.test_f1(labels, pred) self.test_recall(labels, pred) self.test_precision(labels, pred) @@ -528,7 +528,7 @@ class IcingIntensityNN: self.test_loss(t_loss) self.test_accuracy(labels, pred) self.test_auc(labels, pred) - self.test_f1(labels, pred) + #self.test_f1(labels, pred) self.test_recall(labels, pred) self.test_precision(labels, pred) @@ -585,7 +585,7 @@ class IcingIntensityNN: tf.summary.scalar('loss_val', self.test_loss.result(), step=step) tf.summary.scalar('acc_val', self.test_accuracy.result(), step=step) tf.summary.scalar('auc_val', self.test_auc.result(), step=step) - tf.summary.scalar('f1_val', self.test_f1.result(), step=step) + # tf.summary.scalar('f1_val', self.test_f1.result(), step=step) tf.summary.scalar('recall_val', self.test_recall.result(), step=step) tf.summary.scalar('prec_val', self.test_precision.result(), step=step) tf.summary.scalar('num_train_steps', step, step=step)