From beedd21691c979269addebc3ec00c6853881e05a Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Tue, 18 Feb 2025 20:00:48 +0100 Subject: [PATCH] refactor(reforcexy): use model file name if possible MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- ReforceXY/user_data/config-template.json | 1 - ReforceXY/user_data/freqaimodels/ReforceXY.py | 5 +++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ReforceXY/user_data/config-template.json b/ReforceXY/user_data/config-template.json index 98e93f6..cefcccf 100644 --- a/ReforceXY/user_data/config-template.json +++ b/ReforceXY/user_data/config-template.json @@ -130,7 +130,6 @@ ], "label_period_candles": 100, "include_shifted_candles": 6, - "DI_threshold": 10, "weight_factor": 0.9, "principal_component_analysis": false, "use_SVM_to_remove_outliers": false, diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 0f40489..5de1acc 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -366,9 +366,10 @@ class ReforceXY(BaseReinforcementLearningModel): time_spent = time.time() - start self.dd.update_metric_tracker("fit_time", time_spent, dk.pair) - if Path(dk.data_path / "best_model.zip").is_file(): + model_filename = dk.model_filename if dk.model_filename else "best" + if Path(dk.data_path / f"{model_filename}_model.zip").is_file(): logger.info("Callback found a best model.") - best_model = self.MODELCLASS.load(dk.data_path / "best_model") + best_model = self.MODELCLASS.load(dk.data_path / f"{model_filename}_model") return best_model logger.info("Couldn't find best model, using final model instead.") -- 2.43.0