Inference

src/data/lagrangian_datatools.py

@@ -14,6 +14,14 @@ class StandardScaler(object):
     def __call__(self, sample):
         return (sample - self.mean) / self.std
     
+class ScaleDiffusionRange(object):
+    """Scales data to be in range [-1,1]"""
+    def __init__(self):
+        pass
+
+    def __call__(self, sample):
+        return (sample * 2) - 1
+
 class ParticleDataset(Dataset):
     def __init__(self, path, transform=None):