Source code for stonesoup.metricgenerator.base

from abc import abstractmethod

from ..base import Base


[docs] class MetricGenerator(Base): """Metric Generator base class Generates :class:`~.Metric` objects used to assess the performance of a tracker using data held in a :class:`~.MetricManager` object """ @staticmethod def _get_data(manager, key): return manager.states_sets[key]
[docs] @abstractmethod def compute_metric(self, manager, **kwargs): """Compute metric Parameters ---------- manager : MetricManager containing the data to be used to create the metric(s) Returns ------- : list of :class:`~.Metric` objects Generated metrics """ raise NotImplementedError
[docs] class MetricManager(Base): """Metric Manager base class Holds the data and manages the production of :class:`~.Metric` objects through a :class:`~.MetricGenerator` """
[docs] class MetricTableGenerator(MetricGenerator): """Metric Table base class Takes a set of :class: `~.Metric` objects and outputs a table of the values next to a description and target for each metric"""
[docs] @abstractmethod def compute_metric(self, **kwargs): """Generate table Parameters ---------- : set of :class: `~.Metric` objects Returns ------- matplotlib.Table object""" raise NotImplementedError
[docs] class PlotGenerator(MetricGenerator): """PlotGenerator base class PlotGenerators generate metrics that are visualisations. Return :class:`~.PlottingMetric` objects. """
[docs] @abstractmethod def compute_metric(self, **kwargs): """Generate table Parameters ---------- : set of :class: `~.Metric` objects Returns ------- matplotlib.Table object""" raise NotImplementedError