Fixed sparse keyword argument for onehot encoder

shankarpandala · Mar 2, 2024 · 7343eab · 7343eab
1 parent 004eccf
commit 7343eab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lazypredict/Supervised.py b/lazypredict/Supervised.py
@@ -95,7 +95,7 @@
 categorical_transformer_low = Pipeline(
     steps=[
         ("imputer", SimpleImputer(strategy="constant", fill_value="missing")),
-        ("encoding", OneHotEncoder(handle_unknown="ignore", sparse=False)),
+        ("encoding", OneHotEncoder(handle_unknown="ignore", sparse_output=False)),
     ]
 )