diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 58ad8fea9d706f42de12c8d0a43486dce04d82d9..489b2887cf62ab04f9befc19945a4093ac85e926 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