fix swapped labels and probs

michaelbyun · Apr 16, 2023 · fdaeea1 · fdaeea1
1 parent b702eda
commit fdaeea1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elk/calibration.py b/elk/calibration.py
@@ -34,8 +34,8 @@ def update(self, labels: Tensor, probs: Tensor) -> "CalibrationError":
  assert labels.shape == probs.shape
  assert torch.is_floating_point(probs)
 
- self.labels.append(probs)
- self.pred_probs.append(labels)
+ self.labels.append(labels)
+ self.pred_probs.append(probs)
  return self
 
  def compute(self, p: int = 2) -> CalibrationEstimate: