diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 7b20911736ece7f9f867ded0039de16abff63c05..49ef4a3309a93bf7f69152fd16ea282cff540c7c 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -535,14 +535,19 @@ class IcingIntensityNN: self.test_loss.reset_states() self.test_accuracy.reset_states() + self.test_auc.reset_states() + self.test_recall.reset_states() + self.test_precision.reset_states() + for data0, label in self.test_dataset: ds = tf.data.Dataset.from_tensor_slices((data0, label)) ds = ds.batch(BATCH_SIZE) for mini_batch in ds: self.test_step(mini_batch) - print('loss, acc : ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy()) - print('---------------------------------------------------------') + print('loss, acc, auc, recall, precision: ', 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()) + print('--------------------------------------------------') ckpt_manager.save() if self.DISK_CACHE and epoch == 0: