cellpin.pl.losses

Contents

cellpin.pl.losses#

cellpin.pl.losses(log_path, keys=None, *, smooth=0, figsize=None, save=None)#

Plot validation loss curves from a Lightning CSVLogger metrics.csv.

Return type:

None

Args:

log_path: Path to a metrics.csv file or a directory containing one. keys: Column names to plot, e.g. ["val_loss", "val_reconst_loss"].

Defaults to ["val_loss", "val_reconst_loss", "val_inv_loss"]. Pass "all" to show every available val loss.

smooth: Width of a centered rolling-mean window. 0 (default)

disables smoothing.

figsize: (width, height) in inches. Auto-sized from the number of

panels when omitted.

save: If given, the figure is saved to this path at 300 dpi.