From 8c3ae4886e712061f9f696ad378686de1c051cb8 Mon Sep 17 00:00:00 2001 From: EdenWuyifan Date: Sat, 1 Jun 2024 22:20:11 -0400 Subject: [PATCH] tmp --- alpha_automl/hyperparameter_tuning/smac.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/alpha_automl/hyperparameter_tuning/smac.py b/alpha_automl/hyperparameter_tuning/smac.py index 9fe1489..6bee524 100644 --- a/alpha_automl/hyperparameter_tuning/smac.py +++ b/alpha_automl/hyperparameter_tuning/smac.py @@ -99,7 +99,12 @@ def gen_configspace(pipeline): configspace = ConfigurationSpace(seed=0) all_params = {} for primitive, prim_obj in pipeline.steps: - step_type = PRIMITIVE_TYPES[primitive] + if "feature_engine.creation" in primitive: + step_type = "FEATURE_GENERATOR" + elif "feature_engine.selection" in primitive: + step_type = "FEATURE_SELECTOR" + else: + step_type = PRIMITIVE_TYPES[primitive] try: params = SMAC_DICT[primitive] add_params(params, all_params)