diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py
index 31f3beca505200719f4184c8ff914685035a91a8..a3a85cda9c3cf3f5294f8f475781a02a8d8940fa 100644
--- a/modules/deeplearning/icing.py
+++ b/modules/deeplearning/icing.py
@@ -139,6 +139,9 @@ class IcingIntensityNN:
         self.train_accuracy = None
         self.test_loss = None
         self.test_accuracy = None
+        self.test_auc = None
+        self.test_recall = None
+        self.test_precision = None
 
         self.learningRateSchedule = None
         self.num_data_samples = None
@@ -204,6 +207,12 @@ class IcingIntensityNN:
         label = np.where(label == -1, 0, label)
 
         # binary, two class
+        # label = np.where(label != 0, 1, label)
+        # label = label.reshape((label.shape[0], 1))
+
+        keep = (label == 0) | (label == 3) | (label == 4) | (label == 5) | (label == 6)
+        data = data[keep,]
+        label = label[keep]
         label = np.where(label != 0, 1, label)
         label = label.reshape((label.shape[0], 1))
 
@@ -444,6 +453,9 @@ 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)
 
     def predict(self, mini_batch):
         inputs = [mini_batch[0]]