snapshot...

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: