diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 964d03b16af1532965a83a41a82118a200088a4c..c659876b61b35b81bf1bc9fce5e23c30344253ff 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -1222,7 +1222,7 @@ def run_mean_std(check_cloudy=False, no_icing_to_icing_ratio=5): return mean_dct, std_dct -def split_data(num_obs, perc=0.2, skip=1): +def split_data(num_obs, perc=0.2, skip=1, shuffle=True): idxs = np.arange(num_obs) idxs = list(idxs) @@ -1237,10 +1237,10 @@ def split_data(num_obs, perc=0.2, skip=1): test_idxs = np.array(test_idxs) train_idxs = np.array(train_idxs) - #np.random.shuffle(test_idxs) - #np.random.shuffle(train_idxs) + if shuffle: + np.random.shuffle(test_idxs) + np.random.shuffle(train_idxs) - #return np.sort(train_idxs[::skip]), np.sort(test_idxs[::skip]) return train_idxs[::skip], test_idxs[::skip]