def compute(self) -> Tensor:
"""
Computes AUROC based on inputs passed in to ``update`` previously.
"""
preds = dim_zero_cat(self.preds)
target = dim_zero_cat(self.target)
return _auroc_compute(
preds,
target,
self.mode,
self.num_classes,
self.pos_label,
self.average,
self.max_fpr,
)
def compute(self) -> Tensor:
"""Computes AUROC based on inputs passed in to ``update`` previously."""
if not self.mode:
raise RuntimeError("You have to have determined mode.")
preds = dim_zero_cat(self.preds)
target = dim_zero_cat(self.target)
return _auroc_compute(
preds,
target,
self.mode,
self.num_classes,
self.pos_label,
self.average,
self.max_fpr,
)
