snapshot...

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: