learning_curve

Contents

learning_curve#

class braintools.visualize.learning_curve(train_sizes, train_scores, validation_scores, ax=None, figsize=(10, 6), title='Learning Curve', **kwargs)#

Plot learning curve showing training and validation performance.

Parameters:

  • train_sizes (ndarray) – Training set sizes.

  • train_scores (ndarray) – Training scores (can be 2D for multiple runs).

  • validation_scores (ndarray) – Validation scores (can be 2D for multiple runs).

  • ax (Axes | None) – Axes to plot on.

  • figsize (Tuple[float, float]) – Figure size if creating new figure.

  • title (str) – Plot title.

  • **kwargs – Additional arguments passed to plot.

Returns:

ax – The axes object containing the plot.

Return type:

Axes

Contents