drytorch.trackers.matplotlib

Plotting with matplotlib.

Classes

MatPlotter([model_names, source_names, ...])

Tracker that organizes metrics as subplots using matplotlib.
class MatPlotter(model_names: Iterable[str] = (), source_names: Iterable[str] = (), metric_names: Iterable[str] = (), metric_loader: MetricLoader | None = None, start: int = 1)[source]

Bases: BasePlotter[tuple[Figure, Axes]]

Tracker that organizes metrics as subplots using matplotlib.

Initialize.

Parameters:

  • model_names (Iterable[str]) – the names of the models to plot. Defaults to all.

  • source_names (Iterable[str]) – the names of the sources to plot. Defaults to all.

  • metric_names (Iterable[str]) – the names of the metrics to plot. Defaults to all.

  • metric_loader (MetricLoader | None) – a tracker that can load metrics from a previous run.

  • start (int) – if positive, the epoch from which to start plotting; if negative, the last number of epochs. Defaults to all.

close_all()[source]

Close all figures associated with this plotter.