diff --git a/model_data/simulation_system/training.py b/model_data/simulation_system/training.py index d0a2c8ac..2a01485c 100644 --- a/model_data/simulation_system/training.py +++ b/model_data/simulation_system/training.py @@ -145,6 +145,8 @@ def training( plt.tight_layout() RESIDUAL_FILE = "residuals.png" plt.savefig(output_base / METRICS_FOLDER / RESIDUAL_FILE, dpi=120) + plt.savefig(RESIDUAL_FILE, dpi=120) + # TODO: introduce a seperate script for model optimisation, and from there, optimise for deployment # Imagining for now that the model trained here is the best model amongst all models built