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