train on normalized data

astanziola · Apr 11, 2022 · a2ce8ad · a2ce8ad
1 parent 64c901c
commit a2ce8ad
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/utils.py b/scripts/utils.py
@@ -40,6 +40,10 @@ def __init__(self, path: str):
     self.x = self.x[..., np.newaxis]
     self.y = self.y[..., np.newaxis]
 
+    # Normalize data
+    self.x, _, _ = gaussian_normalize(self.x)
+    self.y, _, _ = gaussian_normalize(self.y)
+
   def __len__(self):
     return self.x.shape[0]