incerto.utils.plot_training_curves

incerto.utils.plot_training_curves#

incerto.utils.plot_training_curves(train_losses, val_losses=None, train_accs=None, val_accs=None, title='Training Curves', save_path=None, show=True)[source]#

Plot training and validation curves.

Parameters:
  • train_losses (List[float]) – Training losses per epoch

  • val_losses (Optional[List[float]]) – Validation losses per epoch (optional)

  • train_accs (Optional[List[float]]) – Training accuracies per epoch (optional)

  • val_accs (Optional[List[float]]) – Validation accuracies per epoch (optional)

  • title (str) – Plot title

  • save_path (Optional[str]) – Path to save figure (optional)

  • show (bool) – Whether to call plt.show() (default: True)