probly.visualization.ood

Plotting utilities for OOD evaluation.

Functions

plot_histogram(id_scores, ood_scores[, ax, ...])

Plot ID vs OOD score histogram.

plot_pr_curve(recall, precision, aupr[, ax])

Plot Precision-Recall curve.

plot_roc_curve(fpr, tpr, auroc[, fpr95, ax])

Plot ROC curve.
probly.visualization.ood.plot_histogram(id_scores, ood_scores, ax=None, bins=50, title='Score Distribution')[source]

Plot ID vs OOD score histogram.

Parameters:

  • id_scores (np.ndarray)

  • ood_scores (np.ndarray)

  • ax (Axes | None)

  • bins (int)

  • title (str)

Return type:

Figure | SubFigure

probly.visualization.ood.plot_pr_curve(recall, precision, aupr, ax=None)[source]

Plot Precision-Recall curve.

Parameters:

  • recall (np.ndarray)

  • precision (np.ndarray)

  • aupr (float)

  • ax (Axes | None)

Return type:

Figure | SubFigure

probly.visualization.ood.plot_roc_curve(fpr, tpr, auroc, fpr95=None, ax=None)[source]

Plot ROC curve.

Parameters:

  • fpr (np.ndarray)

  • tpr (np.ndarray)

  • auroc (float)

  • fpr95 (float | None)

  • ax (Axes | None)

Return type:

Figure | SubFigure