diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py index 6ada5b6c625c81dd628ed4f4cccaf3b3ed520cad..3ee0f26a8d834f4b33fcbfde33e330e6616d9cab 100644 --- a/modules/deeplearning/icing.py +++ b/modules/deeplearning/icing.py @@ -240,7 +240,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=8) + trn_idxs, tst_idxs = split_data(num_obs, skip=4) self.num_data_samples = trn_idxs.shape[0] self.get_train_dataset(trn_idxs) @@ -346,7 +346,7 @@ class IcingIntensityNN: else: flat = self.X_img n_hidden = self.X_img.shape[1] - n_hidden = 40 + n_hidden = 50 fac = 1