diff --git a/expo/datasets.yaml b/expo/datasets.yaml index 45150833a..512cbc292 100644 --- a/expo/datasets.yaml +++ b/expo/datasets.yaml @@ -151,7 +151,7 @@ datasets: \ the target column `Class`.\nPerform data analysis, data preprocessing, feature\ \ engineering, and modeling to predict the target. \nReport f1 weighted on the\ \ eval data. Do not plot or make any visualizations.\n" - concrete-strength: + concrete-strength: dataset: concrete-strength metric: rmse target_col: Strength diff --git a/expo/utils.py b/expo/utils.py index 65701c3ec..9c6295fa9 100644 --- a/expo/utils.py +++ b/expo/utils.py @@ -19,7 +19,9 @@ def load_data_config(file_path="data.yaml"): return data_config +DATASET_CONFIG = load_data_config("datasets.yaml") DATA_CONFIG = load_data_config() +DATA_CONFIG["datasets"].update(DATASET_CONFIG["datasets"]) def get_mcts_logger():