diff options
Diffstat (limited to 'src/training/util.py')
-rw-r--r-- | src/training/util.py | 19 |
1 files changed, 0 insertions, 19 deletions
diff --git a/src/training/util.py b/src/training/util.py deleted file mode 100644 index 132b2dc..0000000 --- a/src/training/util.py +++ /dev/null @@ -1,19 +0,0 @@ -"""Utility functions for training neural networks.""" - - -class RunningAverage: - """Maintains a running average.""" - - def __init__(self) -> None: - """Initializes the parameters.""" - self.steps = 0 - self.total = 0 - - def update(self, val: float) -> None: - """Updates the parameters.""" - self.total += val - self.steps += 1 - - def __call__(self) -> float: - """Computes the running average.""" - return self.total / float(self.steps) |