diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 613eadf3a11180b69ef653274968c44ad22f9293..8aea3c05978b262d75230eb133772ab14d6758c8 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -465,6 +465,12 @@ class IcingIntensityNN: pred = self.model(inputs, training=False) t_loss = self.loss(labels, pred) + self.test_loss(t_loss) + self.test_accuracy(labels, pred) + self.test_auc(labels, pred) + self.test_recall(labels, pred) + self.test_precision(labels, pred) + def do_training(self, ckpt_dir=None): if ckpt_dir is None: