diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index 19baeaf66822e609a4cfed3db5076f3cc79c2e30..c8d2455153349208ddd2e41fefbe0cacb913b1c8 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -448,31 +448,39 @@ def analyze_all(ice_dct, no_ice_dct, neg_ice_dct): continue ice_times_5_6 = np.array(ice_times_5_6) + ice_keys_5_6 = np.array(ice_keys_5_6) print('5_6: ', ice_times_5_6.shape) ice_times_4 = np.array(ice_files_4) + ice_keys_4 = np.array(ice_times_4) print('4: ', ice_times_4.shape) - np.random.shuffle(ice_times_4) + #np.random.shuffle(ice_times_4) #ice_times_4 = ice_times_4[0:5000] ice_times_3 = np.array(ice_files_3) + ice_keys_3 = np.array(ice_keys_3) print('3: ', ice_times_3.shape) - np.random.shuffle(ice_times_3) + #np.random.shuffle(ice_times_3) #ice_times_3 = ice_times_3[0:4000] ice_times_2 = np.array(ice_times_2) + ice_keys_2 = np.array(ice_keys_2) print('2: ', ice_times_2.shape) np.random.shuffle(ice_times_2) - #ice_times_2 = ice_times_2[0:7000] + #ice_times_2 = ice_times_2[0:30000] ice_times_1 = np.array(ice_files_1) + ice_keys_1 = np.array(ice_keys_1) print('1: ', ice_times_1.shape) - np.random.shuffle(ice_times_1) + #np.random.shuffle(ice_times_1) #ice_times_1 = ice_times_1[0:5000] ice_times = np.concatenate([ice_times_5_6, ice_times_1, ice_times_2, ice_times_3, ice_times_4]) + ice_keys = np.concatenate([ice_keys_5_6, ice_keys_1, ice_keys_2, ice_keys_3, ice_keys_4]) uniq_sorted = np.unique(ice_times) + uniq_sorted_keys = np.unique(ice_keys) print(ice_times.shape, uniq_sorted.shape) + print(ice_keys.shape, uniq_sorted_keys.shape) no_ice_times = np.array(no_ice_times) neg_ice_times = np.array(neg_ice_times)