From: Jérôme Benoit Date: Tue, 18 Feb 2025 19:00:48 +0000 (+0100) Subject: refactor(reforcexy): use model file name if possible X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=beedd21691c979269addebc3ec00c6853881e05a;p=freqai-strategies.git refactor(reforcexy): use model file name if possible Signed-off-by: Jérôme Benoit --- 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.")