Correction for KCE

Project-MONAI · Mar 3, 2023 · 0a0b492 · 0a0b492
1 parent bccdfb8
commit 0a0b492
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MetricsReloaded/metrics/calibration_measures.py b/MetricsReloaded/metrics/calibration_measures.py
@@ -238,7 +238,7 @@ def kernel_calibration_error(self):
         Calibration tests in multi-class classification: A unifying framework.
         Advances in Neural Information Processing Systems, 32:12257–12267, 2019.
         """
-        one_hot_ref = one_hot_encode(self.ref)
+        one_hot_ref = one_hot_encode(self.ref, self.pred.shape[1])
         numb_samples = self.pred.shape[0]
         sum_tot = 0
         for i in range(0,numb_samples):