eugene.plot.training_summary¶
- eugene.plot.training_summary(log_path, metric='r2', figsize=(12, 6), save=None, return_axes=False, **kwargs)¶
Plots the training summary from a given training run
Convenience function to plot loss and metric together.
- Parameters:
log_path (str, optional) – Path to tensorboard log directory.
metrics (str or list of str, optional) – Metrics to plot. Should be the string name of the metric used in PL
figsize (tuple, optional) – The size of the figure.
save (str, optional) – The filename to save the figure to.
return_axes (bool, optional) – If True, returns the axes object.
**kwargs – Additional keyword arguments to pass to seaborn.
- Returns:
If return_axes is True, returns the axes object.
- Return type:
None or matplotlib.pyplot.Axes