diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index c659876b61b35b81bf1bc9fce5e23c30344253ff..86699d55747974a33c263bf488a0e6bd1734e4bf 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -1244,6 +1244,21 @@ def split_data(num_obs, perc=0.2, skip=1, shuffle=True): return train_idxs[::skip], test_idxs[::skip] +def normalize(data, param, mean_std_dict): + + if mean_std_dict.get(param) is None: + return data + + mean, std = mean_std_dict.get(param) + data -= mean + data /= std + + not_valid = np.invert(np.isnan(data)) + data[not_valid] = 0 + + return data + + def test(filename, skip=1): h5f = h5py.File(filename, 'r') time = h5f['time']