diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index ba4839049f4ffa5eefd98a547c3ee46903f40b85..3a8a66c6ed89972b15ee6fae16c5bdd3fa31979f 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -422,12 +422,26 @@ def analyze_all(ice_dct, no_ice_dct, neg_ice_dct): continue ice_times_5_6 = np.array(ice_times_5_6) - ice_times = ice_times_1 + ice_files_2 + ice_files_3 + ice_files_4 - ice_times = np.array(ice_times) + ice_times_4 = np.array(ice_files_4) + np.random.shuffle(ice_times_4) + ice_times_4 = ice_times_4[0:5000] + ice_times_3 = np.array(ice_files_3) - ice_times_2 = np.array(ice_files_2) + np.random.shuffle(ice_times_3) + ice_times_3 = ice_times_3[0:4000] + + ice_times_2 = np.array(ice_times_2) + np.random.shuffle(ice_times_2) + ice_times_2 = ice_times_2[0:7000] + ice_times_1 = np.array(ice_files_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]) + uniq_sorted, _, _, _ = np.unique(ice_times) + no_ice_times = np.array(no_ice_times) neg_ice_times = np.array(neg_ice_times)