Update training.py

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

View file

@ -127,7 +127,7 @@ def training(
logger.info("--- Optimising model for deployment ---")
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
# Save this in the model directory