Source code for deep_bottleneck.plotter.base

import matplotlib


[docs]class BasePlotter: """Base class for plotters.""" plotname = ''
[docs] def generate(self, measures_summary): fig = self.plot(measures_summary) filename = f'plots/{self.plotname}.png' fig.savefig(filename, bbox_inches='tight', dpi=300) self.run.add_artifact(filename, name=self.plotname)
[docs] def plot(self, measures_summary) -> matplotlib.figure.Figure: raise NotImplementedError