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

fix config example in pytorch mlp documentation

Browse Source
This commit is contained in:
Yinon Polak
2023-03-28 14:41:25 +03:00
parent 026b6a39a9
commit b795a70102
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -26,7 +26,7 @@ class PyTorchMLPClassifier(BasePyTorchClassifier):
"trainer_kwargs": {
"max_iters": 5000,
"batch_size": 64,
"max_n_eval_batches": None,
"max_n_eval_batches": null,
},
"model_kwargs": {
"hidden_dim": 512,
@@ -49,7 +49,7 @@ class PyTorchMLPClassifier(BasePyTorchClassifier):
"""
User sets up the training and test data to fit their desired model here
:param data_dictionary: the dictionary constructed by DataHandler to hold
all the training and test data/labels.
all the training and test data/labels.
:raises ValueError: If self.class_names is not defined in the parent class.
"""

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

@@ -27,7 +27,7 @@ class PyTorchMLPRegressor(BasePyTorchRegressor):
"trainer_kwargs": {
"max_iters": 5000,
"batch_size": 64,
"max_n_eval_batches": None,
"max_n_eval_batches": null,
},
"model_kwargs": {
"hidden_dim": 512,
@@ -50,7 +50,7 @@ class PyTorchMLPRegressor(BasePyTorchRegressor):
"""
User sets up the training and test data to fit their desired model here
:param data_dictionary: the dictionary constructed by DataHandler to hold
all the training and test data/labels.
all the training and test data/labels.
"""
n_features = data_dictionary["train_features"].shape[-1]