From 0a0b4926796648cdf7fa152a7249eb20a1d6b2f0 Mon Sep 17 00:00:00 2001
From: Carole Sudre <csudre@Caroles-MBP.home>
Date: Fri, 3 Mar 2023 14:48:17 +0000
Subject: [PATCH] Correction for KCE

---
 MetricsReloaded/metrics/calibration_measures.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/MetricsReloaded/metrics/calibration_measures.py b/MetricsReloaded/metrics/calibration_measures.py
index 81ea813..bebed9c 100644
--- 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):