incerto.utils.plot_2d_classification

incerto.utils.plot_2d_classification#

incerto.utils.plot_2d_classification(X, y, model=None, device='cpu', title='Classification', save_path=None, show=True)[source]#

Plot 2D classification data and decision boundary.

Parameters:

  • X (ndarray) – Input features (N, 2)

  • y (ndarray) – Labels (N,)

  • model (Optional[Module]) – Trained model (optional)

  • device (str) – Device for model inference

  • title (str) – Plot title

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

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

Contents