From 28cefa07b797bb0fed210ad18bb55bc3dda5b396 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 18 Mar 2021 12:26:45 -0500
Subject: [PATCH] updates for icing intensity

---
 modules/icing/pirep_goes.py | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index ba483904..3a8a66c6 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)
 
-- 
GitLab