junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

consolidated location of output storage

Browse source
This commit is contained in:
Khalim Conn-Kowlessar 2023-08-25 15:33:29 +01:00
parent d6562bfab9
commit 67fd184ac5
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
model_data/simulation_system/training.py
View file

@ -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():