From a160c239857de6f2ed0125fcdc867449945f4e56 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 20 Sep 2021 12:17:31 -0500
Subject: [PATCH] snapshot...
---
modules/icing/pirep_goes.py | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index 58ad8fea..489b2887 100644
--- a/modules/icing/pirep_goes.py
+++ b/modules/icing/pirep_goes.py
@@ -1537,6 +1537,18 @@ def split_data(times):
test_time_idxs = np.concatenate(test_time_idxs, axis=None)
train_time_idxs = time_idxs[np.in1d(time_idxs, test_time_idxs, invert=True)]
+ # Keep out
+ out_idxs = []
+ for k, t_rng in enumerate(time_ranges):
+ t_a = time_ranges[k][0]
+ t_b = time_ranges[k][1]
+ tidxs = np.searchsorted(times, [t_a - 10800, t_a])
+ out_idxs.append(np.arange(tidxs[0], tidxs[1], 1))
+ tidxs = np.searchsorted(times, [t_b, t_b + 10800])
+ out_idxs.append(np.arange(tidxs[0], tidxs[1], 1))
+ out_idxs = np.concatenate(out_idxs, axis=None)
+ train_time_idxs = train_time_idxs[np.in1d(train_time_idxs, out_idxs, invert=True)]
+
return train_time_idxs, test_time_idxs
--
GitLab