minor...

modules/util/util.py

@@ -324,4 +324,30 @@ def make_times(dt_str_0, dt_str_1=None, format_code='%Y-%m-%d_%H:%M', num_steps=
 
 def make_histogram(values, edges):
     h = np.histogram(values, bins=edges)
-    return h
+    return h
\ No newline at end of file
+
+
+def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=1.0, seed=None):
+
+    if mean_std_dict.get(param) is None:
+        return data
+
+    shape = data.shape
+    data = data.flatten()
+
+    mean, std, lo, hi = mean_std_dict.get(param)
+    data -= mean
+    data /= std
+
+    if add_noise:
+        if seed is not None:
+            np.random.seed(seed)
+        rnd = np.random.normal(loc=0, scale=noise_scale, size=data.size)
+        data += rnd
+
+    not_valid = np.isnan(data)
+    data[not_valid] = 0
+
+    data = np.reshape(data, shape)
+
+    return data
\ No newline at end of file