Skip to content
Snippets Groups Projects

Inference

Merged Maria Guaranda-Cabezas requested to merge inference into main
3 files
+ 11
2
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -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):
Loading