From 814524919785d0da3bcdb06f1de4a257b9c31992 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 15 Apr 2021 13:39:51 -0500
Subject: [PATCH] keep only < 273 for both icing and no-icing regions
---
modules/icing/pirep_goes.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index 2a2a1e15..169220c5 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))
--
GitLab