diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 6679ff74ad93ef6010bdc74d480264f3d12d71b2..54ace0c9be3c99dd34e3dfc04743088e043ffd73 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -1308,7 +1308,7 @@ def run_mean_std_2(check_cloudy=False, no_icing_to_icing_ratio=5, params=train_p # return mean_std_dct -def split_data(num_obs, perc=0.2, skip=1, shuffle=True): +def split_data(num_obs, perc=0.2, skip=1, shuffle=True, seed=None): idxs = np.arange(num_obs) idxs = list(idxs) @@ -1323,6 +1323,9 @@ def split_data(num_obs, perc=0.2, skip=1, shuffle=True): test_idxs = np.array(test_idxs) train_idxs = np.array(train_idxs) + if seed is not None: + np.random.seed(seed) + if shuffle: np.random.shuffle(test_idxs) np.random.shuffle(train_idxs)