Skip to content

Loss

Source

Plotting Losses.

plot_loss(training_report, fig_size=(8, 5.5), save_path=None, plot_metrics=True, plot_learning_rate=True)

Plot training and validation losses.

Parameters:

  • training_report (TrainingReport) –

    Dataclass containing train and validation loss, learning rate, and any discovered validation metrics.

  • fig_size (tuple[float, float], default: None ) –

    Size of the figure to be plotted. If None, then default settings are applied depending on plot_metrics and plot_learning_rate.

  • save_path (Path | str | None, default: None ) –

    Path to save the figure. If None, the figure will not be saved.

  • plot_metrics (bool, default: True ) –

    Whether to plot the metrics.

  • plot_learning_rate (bool, default: True ) –

    Whether to plot the learning rate.