snapshot..

modules/util/util.py

@@ -435,13 +435,14 @@ def make_histogram(values, edges):
     return h
 
 
-def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=0.01, seed=None, copy=True):
-    if copy:
-        data = data.copy()
+def normalize(data, param, mean_std_dict, copy=True):
 
     if mean_std_dict.get(param) is None:
         return data
 
+    if copy:
+        data = data.copy()
+
     shape = data.shape
     data = data.flatten()
 
@@ -449,15 +450,26 @@ def normalize(data, param, mean_std_dict, add_noise=False, noise_scale=0.01, see
     data -= mean
     data /= std
 
-    if add_noise:
+    not_valid = np.isnan(data)
+    data[not_valid] = 0
+
+    data = np.reshape(data, shape)
+
+    return data
+
+
+def add_noise(data, noise_scale=0.01, seed=None, copy=True):
+    if copy:
+        data = data.copy()
+
+    shape = data.shape
+    data = data.flatten()
+
     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