adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

type hints fixes

Browse Source
This commit is contained in:
Yinon Polak
2023-03-20 20:22:28 +02:00
parent c06cd38951
commit a4b617e482
2 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py
View File

@@ -49,8 +49,8 @@ class PyTorchMLPRegressor(PyTorchRegressor):
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.model_kwargs: Dict[str, Any] = config.get("model_kwargs", model_kwargs)
self.trainer_kwargs: Dict[str, Any] = config.get("trainer_kwargs", trainer_kwargs)
def fit(self, data_dictionary: Dict, dk: FreqaiDataKitchen, **kwargs) -> Any:
"""