Don't remove empty features during imputation

VIDA-NYU · Feb 18, 2024 · 7297a55 · 7297a55
1 parent ed2d87f
commit 7297a55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alpha_automl/pipeline_synthesis/pipeline_builder.py b/alpha_automl/pipeline_synthesis/pipeline_builder.py
@@ -44,7 +44,7 @@ def change_default_hyperparams(primitive_object):
     elif isinstance(primitive_object, OrdinalEncoder):
         primitive_object.set_params(handle_unknown='use_encoded_value', unknown_value=-1)
     elif isinstance(primitive_object, SimpleImputer):
-        primitive_object.set_params(strategy='most_frequent')
+        primitive_object.set_params(strategy='most_frequent', keep_empty_features=True)
 
 
 class BaseBuilder: