small issue in _prepare (didn't return the one-hot encoded values if …

…the one_hot_encoder is fitted)
MaxHalford · Sep 3, 2024 · 77e0603 · 77e0603
1 parent 37e0f59
commit 77e0603
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/prince/mca.py b/prince/mca.py
@@ -61,7 +61,8 @@ def _prepare(self, X):
  return X_enc
  else:
  X_enc = self.one_hot_encoder.transform(X)
- X = pd.DataFrame(X_enc, columns=self.one_hot_encoder.get_feature_names_out(X.columns))
+ X_enc = pd.DataFrame(X_enc, columns=self.one_hot_encoder.get_feature_names_out(X.columns))
+ return X_enc
  return X
 
  @utils.check_is_dataframe_input