Merge pull request #79 from kaseris/metrics

Metric list

src/skelcast/metrics/metric.py

@@ -1,4 +1,6 @@
 from abc import ABC, abstractmethod
+from skelcast.metrics import METRICS
+
 
 class Metric(ABC):
  @abstractmethod
@@ -28,3 +30,24 @@ def reset(self):
  Resets the metric state.
  """
  pass
+
+
+def create_metric(metric_name, **kwargs):
+ metric_class = METRICS.get_module(metric_name)
+ return metric_class(**kwargs)
+
+
+class MetricList:
+ def __init__(self, metric_names):
+ self.metrics = [create_metric(name) for name in metric_names]
+
+ def update(self, output):
+ for metric in self.metrics:
+ metric.update(output)
+
+ def compute(self):
+ return {type(metric).__name__: metric.result() for metric in self.metrics}
+
+ def reset(self):
+ for metric in self.metrics:
+ metric.reset()