|
|
|
from collections import OrderedDict |
|
|
|
import numpy as np |
|
|
|
|
|
class LogBuffer: |
|
|
|
def __init__(self): |
|
self.val_history = OrderedDict() |
|
self.n_history = OrderedDict() |
|
self.output = OrderedDict() |
|
self.ready = False |
|
|
|
def clear(self): |
|
self.val_history.clear() |
|
self.n_history.clear() |
|
self.clear_output() |
|
|
|
def clear_output(self): |
|
self.output.clear() |
|
self.ready = False |
|
|
|
def update(self, vars, count=1): |
|
assert isinstance(vars, dict) |
|
for key, var in vars.items(): |
|
if key not in self.val_history: |
|
self.val_history[key] = [] |
|
self.n_history[key] = [] |
|
self.val_history[key].append(var) |
|
self.n_history[key].append(count) |
|
|
|
def average(self, n=0): |
|
"""Average latest n values or all values.""" |
|
assert n >= 0 |
|
for key in self.val_history: |
|
values = np.array(self.val_history[key][-n:]) |
|
nums = np.array(self.n_history[key][-n:]) |
|
avg = np.sum(values * nums) / np.sum(nums) |
|
self.output[key] = avg |
|
self.ready = True |
|
|
