tomodrgn.analysis.plot_losses

plot_losses(runlog: str) → tuple[Figure, Axes]

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

Parameters:

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

Returns:

matplotlib figure and axes of line plots for each loss