snapshot...

modules/deeplearning/icing_cnn.py

@@ -474,6 +474,7 @@ class IcingIntensityNN:
 
         self.test_loss(t_loss)
         self.test_accuracy(labels, pred)
+        if NumClasses == 2:
             self.test_auc(labels, pred)
             self.test_recall(labels, pred)
             self.test_precision(labels, pred)