diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py index 42f5a1a14d0e9ebf21a88557c64e360b21e19e65..99b3908b873f12db2cbeebb750b610fb7bf3eba1 100644 --- a/modules/deeplearning/icing.py +++ b/modules/deeplearning/icing.py @@ -527,7 +527,7 @@ class IcingIntensityNN: for mini_batch in ds: self.test_step(mini_batch) - print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result()) + print('loss, acc, auc: ', self.test_loss.result().numpy(), self.test_accuracy.result().numpy()) ckpt_manager.save() if self.DISK_CACHE and epoch == 0: