tomodrgn.analysis.parse_all_losses

parse_all_losses(run_log: str) → tuple[ndarray, ndarray, ndarray]

Parse MSE, KLD, and total loss at each epoch from run.log output.

Parameters:

run_log – Path to run.log output file.

Returns:

tuple of arrays for MSE loss, KLD loss, and total loss, at each epoch.