Method compute

metrics/rce.py:169–181

Compute and return the accumulated rce.

(self)

    )

  def compute(self) -> torch.Tensor:
    """
    Compute and return the accumulated rce.
    """
    baseline_mean = self.mean_label.compute()

    baseline_ce = _binary_cross_entropy_with_clipping(
      baseline_mean, baseline_mean, reduction="mean", epsilon=self.epsilon
    )

    pred_ce = self.binary_cross_entropy.compute()

    return (1.0 - (pred_ce / baseline_ce)) * 100

  def reset(self):
    """