add scoring metric as metafeature

VIDA-NYU · Apr 25, 2024 · 5d4ed25 · 5d4ed25
1 parent 1d9810e
commit 5d4ed25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alpha_automl/pipeline_synthesis/setup_search.py b/alpha_automl/pipeline_synthesis/setup_search.py
@@ -104,7 +104,7 @@ def update_config(task_name, metric, grammar, metadata):
     config['METRIC'] = metric
     config['DATASET'] = f'DATASET_{task_name}'
     config['GRAMMAR'] = grammar
-    metafeatures = compute_metafeatures(metadata)
+    metafeatures = compute_metafeatures(metric, metadata)
     config['DATASET_METAFEATURES'] = metafeatures + [0] * (8 - len(metafeatures))
 
     return config