From f67395cc0712dd3b3ddff8c1e573bf8826350137 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Thu, 15 Apr 2021 15:00:51 -0500 Subject: [PATCH] snapshot... --- modules/icing/pirep_goes.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 964d03b1..c659876b 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] -- GitLab