From 15aa0778bb2552de13ae3efc40d75729b1d54a00 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 15 May 2021 11:56:04 -0500
Subject: [PATCH] minor...

---
 modules/icing/pirep_goes.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py
index bde3c47a..5fada8e0 100644
--- a/modules/icing/pirep_goes.py
+++ b/modules/icing/pirep_goes.py
@@ -990,7 +990,21 @@ def tile_extract(trnfile='/home/rink/tiles_l1b_train.h5', tstfile='/home/rink/ti
     for ds_name in train_params:
         data_dct[ds_name] = np.concatenate([icing_data_dct[ds_name], no_icing_data_dct[ds_name]])
 
-    trn_idxs, tst_idxs = split_data(icing_intensity.shape[0], shuffle=False, perc=split)
+    # do sort -------------------------------------
+    ds_indexes = np.argsort(icing_times)
+    for ds_name in train_params:
+        data_dct[ds_name] = data_dct[ds_name][ds_indexes]
+    icing_intensity = icing_intensity[ds_indexes]
+    icing_times = icing_times[ds_indexes]
+    icing_lons = icing_lons[ds_indexes]
+    icing_lats = icing_lats[ds_indexes]
+
+    #trn_idxs, tst_idxs = split_data(icing_intensity.shape[0], shuffle=False, perc=split)
+    all_idxs = np.arange(icing_intensity.shape[0])
+    splt_idx = int(icing_intensity.shape[0] * 0.8)
+    trn_idxs = all_idxs[0:splt_idx]
+    tst_idxs = all_idxs[splt_idx:]
+    # ---------------------------------------------
 
     trn_data_dct = {}
     for ds_name in train_params:
-- 
GitLab