From 270da37287498414b80f2f0ca1cd1b52fe2607c5 Mon Sep 17 00:00:00 2001 From: Ines Oliveira e Silva Date: Thu, 29 Feb 2024 10:24:59 +0000 Subject: [PATCH] Remove label from feature importance suppression (#181) --- .../methods/preprocessing/feature_importance_suppression.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py b/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py index 860cabeb..b6a888b6 100644 --- a/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py +++ b/src/aequitas/flow/methods/preprocessing/feature_importance_suppression.py @@ -71,8 +71,7 @@ def fit(self, X: pd.DataFrame, y: pd.Series, s: Optional[pd.Series]) -> None: n_estimators=self.n_estimators, random_state=self.seed, n_jobs=self.n_jobs ) - features = pd.concat([X, y], axis=1) - features = pd.get_dummies(features) + features = pd.get_dummies(X) target = s.copy() features_train, features_val, target_train, target_val = train_test_split(