summaryrefslogtreecommitdiff
path: root/config/lstm-xreg.py
diff options
context:
space:
mode:
Diffstat (limited to 'config/lstm-xreg.py')
-rw-r--r--config/lstm-xreg.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/config/lstm-xreg.py b/config/lstm-xreg.py
index 66f7c51..f8c5094 100644
--- a/config/lstm-xreg.py
+++ b/config/lstm-xreg.py
@@ -14,10 +14,10 @@ seq_div_size = 200
layers = [
{'dim': 1024,
- 'xreg': (768, 0.1, 10, 10, 6)
+ 'xreg': (768, 0.1, 10, 10, 10, 2)
},
{'dim': 1024,
- 'xreg': (768, 0.1, 10, 10, 6)
+ 'xreg': (768, 0.1, 10, 10, 10, 5)
},
{'dim': 1024,
},