From 07703920b519572dfcd87a95651b73556cc6bf04 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Tue, 13 Apr 2021 17:24:35 -0500 Subject: [PATCH] keep more data in process_2 --- modules/icing/pirep_goes.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 32a7b695..cddb554c 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -419,7 +419,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct): print('2: ', ice_keys_2.shape) np.random.seed(42) np.random.shuffle(ice_keys_2) - ice_keys_2 = ice_keys_2[0:30000] + ice_keys_2 = ice_keys_2[0:60000] ice_keys_1 = np.array(ice_keys_1) print('1: ', ice_keys_1.shape) @@ -437,7 +437,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct): print('no ice total: ', no_ice_keys.shape) np.random.seed(42) np.random.shuffle(no_ice_keys) - no_ice_keys = no_ice_keys[0:50000] + no_ice_keys = no_ice_keys[0:150000] uniq_sorted_no_ice = np.unique(no_ice_keys) print('no ice: ', no_ice_keys.shape, uniq_sorted_no_ice.shape) @@ -449,7 +449,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct): print('neg ice total: ', neg_ice_keys.shape) np.random.seed(42) np.random.shuffle(neg_ice_keys) - neg_ice_keys = neg_ice_keys[0:5000] + neg_ice_keys = neg_ice_keys[0:10000] uniq_sorted_neg_ice = np.unique(neg_ice_keys) print('neg ice: ', neg_ice_keys.shape, uniq_sorted_neg_ice.shape) -- GitLab