flambe.metric.dev.auc

Module Contents

class flambe.metric.dev.auc.AUC(max_fpr=1.0)[source]

Bases: flambe.metric.metric.Metric

compute(self, pred: torch.Tensor, target: torch.Tensor)[source]

Compute AUC at the given max false positive rate.

Parameters:
  • pred (torch.Tensor) – The model predictions
  • target (torch.Tensor) – The binary targets
Returns:

The computed AUC

Return type:

torch.Tensor