Iter on suggestions

INRIA · Oct 26, 2023 · f7a1cde · f7a1cde
1 parent cdfc229
commit f7a1cde
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/python_scripts/linear_models_feature_engineering_classification.py b/python_scripts/linear_models_feature_engineering_classification.py
@@ -235,7 +235,10 @@ def plot_decision_boundary(model, title=None):
 # %%
 from sklearn.preprocessing import KBinsDiscretizer
 
-classifier = make_pipeline(KBinsDiscretizer(n_bins=5), LogisticRegression())
+classifier = make_pipeline(
+    KBinsDiscretizer(n_bins=5, encode="onehot"),  # already the default params
+    LogisticRegression(),
+)
 classifier
 
 # %%
@@ -290,8 +293,8 @@ def plot_decision_boundary(model, title=None):
 #
 # ```{note}
 # Notice that `KBinsDiscretizer(encode="onehot")` and `SplineTransformer` do not
-# require additional scaling. Indeed, they can replace the scaling step for numerical
-# features: they both create features with values in the [0, 1] range.
+# require additional scaling. Indeed, they can replace the scaling step for
+# numerical features: they both create features with values in the [0, 1] range.
 # ```
 
 # %% [markdown]