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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -26,7 +26,7 @@ class PyTorchMLPClassifier(BasePyTorchClassifier):
"trainer_kwargs": { "trainer_kwargs": {
"max_iters": 5000, "max_iters": 5000,
"batch_size": 64, "batch_size": 64,
"max_n_eval_batches": None, "max_n_eval_batches": null,
}, },
"model_kwargs": { "model_kwargs": {
"hidden_dim": 512, "hidden_dim": 512,

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

@ -27,7 +27,7 @@ class PyTorchMLPRegressor(BasePyTorchRegressor):
"trainer_kwargs": { "trainer_kwargs": {
"max_iters": 5000, "max_iters": 5000,
"batch_size": 64, "batch_size": 64,
"max_n_eval_batches": None, "max_n_eval_batches": null,
}, },
"model_kwargs": { "model_kwargs": {
"hidden_dim": 512, "hidden_dim": 512,