diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 32a7b695068d3550640dedaa6edac091ace5d197..cddb554c338e59ae04a3c251707684aa4653a591 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)