diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 2a2a1e1532bbd19059c7cfa7e27499bde7fa5b39..169220c50a26a84fb2826555c5d364cf1b902aac 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -1243,12 +1243,12 @@ def split_data(num_obs, perc=0.2, skip=1): return np.sort(train_idxs[::skip]), np.sort(test_idxs[::skip]) -def test(filename): +def test(filename, skip=1): h5f = h5py.File(filename, 'r') time = h5f['time'] intsty = h5f['icing_intensity'] - trn_idxs, tst_idxs = split_data(time.shape[0]) + trn_idxs, tst_idxs = split_data(time.shape[0], skip=skip) print(np.histogram(intsty[trn_idxs], bins=7)) print(np.histogram(intsty[tst_idxs], bins=7))