pdft.viz.plot_training_loss

pdft.viz.plot_training_loss#

pdft.viz.plot_training_loss(history, *, output_path=None, title='Training loss', smooth_alpha=None)[source]#

Plot a single loss trajectory (optionally smoothed) and return the Figure.

If output_path is given, saves the figure to that path and returns it.

Parameters: