probly.metrics.torch

PyTorch implementation of Metrics.

Functions

auc_torch	Compute area under a curve using the trapezoid rule.
average_precision_score_torch	Compute average precision for PyTorch tensors.
precision_recall_curve_torch	Compute precision-recall curve along the last axis.
roc_auc_score_torch	Compute area under the ROC curve for PyTorch tensors.
roc_curve_torch	Compute ROC curve along the last axis.