Remove label from feature importance suppression (#181)

dssg · Feb 29, 2024 · 270da37 · 270da37
1 parent b922a4e
commit 270da37
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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(