fixed xgboost model yaml to load imputer and normalization from the m…

…odel_params which optuna can modify
mmcdermott · Aug 21, 2024 · e6cf085 · e6cf085
1 parent ecf9292
commit e6cf085
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/MEDS_tabular_automl/configs/model/xgboost.yaml b/src/MEDS_tabular_automl/configs/model/xgboost.yaml
@@ -9,8 +9,8 @@ model_target:
   output_filepath: ${output_filepath}
   log_dir: ${log_dir}
   cache_dir: ${cache_dir}
-  imputer: ${imputer}
-  normalization: ${normalization}
+  imputer: ${model_params.iterator.imputer}
+  normalization: ${model_params.iterator.normalization}
   # tabularization: ${tabularization} # Ideally we should define tabularization here, but there is an issue initializing with it's resolvers.
 
 model_params: