From 67fd184ac570824a56406d14462e28e37e126f29 Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Fri, 25 Aug 2023 15:33:29 +0100 Subject: [PATCH] consolidated location of output storage --- model_data/simulation_system/training.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/model_data/simulation_system/training.py b/model_data/simulation_system/training.py index 561d1e1d..4d751c9b 100644 --- a/model_data/simulation_system/training.py +++ b/model_data/simulation_system/training.py @@ -102,7 +102,7 @@ def training( if model_type == "autogluon": model_root = f"{target_column}-{hyperparameters['presets']}-{hyperparameters['time_limit']}-{TIMESTAMP}".lower() - output_base = Path(MODEL_DIRECTORY) / target_column / model_type / model_root + output_base = BASE_REGISTRY_PATH / target_column / model_type / model_root model = AutogluonModel( output_filepath=output_base / MODEL_FOLDER @@ -164,7 +164,6 @@ def training( # TODO: Need a model registry - for now have this as a CSV # Save this in the model directory logger.info("--- Append registry with new model ---") - registry_path = BASE_REGISTRY_PATH / target_column / REGISTRY_FILE if registry_path.exists():