tomodrgn.convergence.plot_loss

plot_loss(runlog: str, outdir: str, plot_format: Literal['png', 'svg']) → None

Plot the total loss, reconstruction loss, and KLD divergence per epoch.

Parameters:

• runlog – the run.log auto-generated by tomodrgn train_vae

• outdir – path to base directory to save outputs

• plot_format – file format with which to save plots

Returns:

None