From cd00bfa0da85431816d6ad81b0bf2cf19a945951 Mon Sep 17 00:00:00 2001 From: achamma Date: Tue, 4 Jun 2024 00:42:57 +0200 Subject: [PATCH] Fix DNN multi_class --- hidimstat/BBI.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/hidimstat/BBI.py b/hidimstat/BBI.py index a5fa9a5..4768aff 100644 --- a/hidimstat/BBI.py +++ b/hidimstat/BBI.py @@ -764,6 +764,9 @@ def compute_importance(self, X=None, y=None): for ind_fold, estimator in enumerate(self.list_estimators): if self.type == "DNN": for y_col in range(y[ind_fold].shape[-1]): + _ = self.estimator.encode_outcome( + self.y_train[ind_fold], train=True + )[y_col] y[ind_fold] = self.estimator.encode_outcome( y[ind_fold], train=False )[y_col]