junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-30 13:10:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update training.py

Browse source
This commit is contained in:
quandanrepo 2023-08-19 14:01:13 +01:00 • committed by GitHub
parent 261a811955
commit f2eab51ff4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
model_data/simulation_system/training.py
View file

@ -127,7 +127,7 @@ def training(
logger.info("--- Optimising model for deployment ---") logger.info("--- Optimising model for deployment ---")
deployment_model_path = model.optimise_model_for_deployment(deployment_path= output_base / DEPLOYMENT_FOLDER) deployment_model_path = model.optimise_model_for_deployment(deployment_path= output_base / DEPLOYMENT_FOLDER)
logger.info("Optimised version of best model can be found at: {deployment_model_path}") logger.info(f"Optimised version of best model can be found at: {deployment_model_path}")
# TODO: Need a model registry - for now have this as a CSV # TODO: Need a model registry - for now have this as a CSV
# Save this in the model directory # Save this in the model directory