diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index da3b44efc906df773194e24614cdbbe7d412e9dc..40af66f3f4e67b402f7ce5adc52dbd8fcfd51452 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -208,12 +208,6 @@ class IcingIntensityNN: 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)) - if CACHE_DATA_IN_MEM: self.in_mem_data_cache[key] = (data, label)