def _roc_update( preds: torch.Tensor, target: torch.Tensor, num_classes: Optional[int] = None, pos_label: Optional[int] = None, ) -> Tuple[torch.Tensor, torch.Tensor, int, int]: return _precision_recall_curve_update(preds, target, num_classes, pos_label)
def update(self, preds: torch.Tensor, target: torch.Tensor): """ Update state with predictions and targets. Args: preds: Predictions from model target: Ground truth values """ preds, target, num_classes, pos_label = _precision_recall_curve_update( preds, target, self.num_classes, self.pos_label) self.preds.append(preds) self.target.append(target) self.num_classes = num_classes self.pos_label = pos_label