snapshot...

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]]