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

type hint init in pytorch mlp classes

Browse Source
This commit is contained in:
Yinon Polak
2023-03-21 15:19:34 +02:00
parent eba82360fa
commit 83a7d888bc
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
View File

@@ -38,7 +38,7 @@ class PyTorchMLPClassifier(PyTorchClassifier):
}
"""
def __init__(self, **kwargs):
def __init__(self, **kwargs) -> None:
super().__init__(**kwargs)
config = self.freqai_info.get("model_training_parameters", {})
self.learning_rate: float = config.get("learning_rate", 3e-4)

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

@@ -39,7 +39,7 @@ class PyTorchMLPRegressor(PyTorchRegressor):
}
"""
def __init__(self, **kwargs):
def __init__(self, **kwargs) -> None:
super().__init__(**kwargs)
config = self.freqai_info.get("model_training_parameters", {})
self.learning_rate: float = config.get("learning_rate", 3e-4)