From f20f00e5fef9e7d007608341ac665c2a58ec2b44 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 30 Apr 2021 09:59:06 -0500
Subject: [PATCH] snapshot...
---
modules/icing/pirep_goes.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index 6679ff74..54ace0c9 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)
--
GitLab