From 9906e7d646f980ebbed60e4f9501042ac802750e Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Tue, 21 Mar 2023 11:23:45 +0200 Subject: [PATCH] clean code --- .../prediction_models/PyTorchMLPClassifier.py | 14 ++++---------- .../prediction_models/PyTorchMLPRegressor.py | 14 ++++---------- 2 files changed, 8 insertions(+), 20 deletions(-) diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py b/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py index 373b81a82..e26b8b52c 100644 --- a/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py +++ b/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py @@ -38,18 +38,12 @@ class PyTorchMLPClassifier(PyTorchClassifier): } """ - def __init__( - self, - learning_rate: float = 3e-4, - model_kwargs: Dict[str, Any] = {}, - trainer_kwargs: Dict[str, Any] = {}, - **kwargs - ): + def __init__(self, **kwargs): super().__init__(**kwargs) config = self.freqai_info.get("model_training_parameters", {}) - self.learning_rate: float = config.get("learning_rate", learning_rate) - self.model_kwargs: Dict[str, Any] = config.get("model_kwargs", model_kwargs) - self.trainer_kwargs: Dict[str, Any] = config.get("trainer_kwargs", trainer_kwargs) + self.learning_rate: float = config.get("learning_rate", 3e-4) + self.model_kwargs: Dict[str, Any] = config.get("model_kwargs", {}) + self.trainer_kwargs: Dict[str, Any] = config.get("trainer_kwargs", {}) def fit(self, data_dictionary: Dict, dk: FreqaiDataKitchen, **kwargs) -> Any: """ diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py b/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py index 16e7c0e79..94c0dfe46 100644 --- a/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py +++ b/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py @@ -39,18 +39,12 @@ class PyTorchMLPRegressor(PyTorchRegressor): } """ - def __init__( - self, - learning_rate: float = 3e-4, - model_kwargs: Dict[str, Any] = {}, - trainer_kwargs: Dict[str, Any] = {}, - **kwargs - ): + def __init__(self, **kwargs): super().__init__(**kwargs) config = self.freqai_info.get("model_training_parameters", {}) - self.learning_rate: float = config.get("learning_rate", learning_rate) - self.model_kwargs: Dict[str, Any] = config.get("model_kwargs", model_kwargs) - self.trainer_kwargs: Dict[str, Any] = config.get("trainer_kwargs", trainer_kwargs) + self.learning_rate: float = config.get("learning_rate", 3e-4) + self.model_kwargs: Dict[str, Any] = config.get("model_kwargs", {}) + self.trainer_kwargs: Dict[str, Any] = config.get("trainer_kwargs", {}) def fit(self, data_dictionary: Dict, dk: FreqaiDataKitchen, **kwargs) -> Any: """