plot_history

ModelComparator.plot_history(values=('loss',), exp_names=None, save_prefix='')[source]

Line plot of values against epochs for different experiments.

Saves plot in ModelComparator.save_dir.

Parameters
  • values (Iterable[str]) – Key in history dict to be plotted.

  • exp_names (Optional[Iterable[str]]) – Compare only a subset of experiments.

  • save_prefix (str) – Plot is saved under {save_prefix}umap_{exp_name}.png.

Return type

Nothing, plots and saves lineplots