snapshot...

modules/deeplearning/icing_cnn.py

@@ -474,9 +474,10 @@ class IcingIntensityNN:
 
         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)
+        if NumClasses == 2:
+            self.test_auc(labels, pred)
+            self.test_recall(labels, pred)
+            self.test_precision(labels, pred)
 
     def predict(self, mini_batch):
         inputs = [mini_batch[0]]