diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 069269a531d99c1ca32095b6d11e734bcdd918e9..9bb4deab887f3f052c3b34f0b0463b48eea94ba1 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -248,7 +248,7 @@ class IcingIntensityNN: self.h5f = h5py.File(filename, 'r') time = self.h5f['time'] num_obs = time.shape[0] - trn_idxs, tst_idxs = split_data(num_obs, skip=4) + trn_idxs, tst_idxs = split_data(num_obs) self.num_data_samples = trn_idxs.shape[0] self.get_train_dataset(trn_idxs)