liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-04 11:30:57 +08:00
Code Activity

Add SFM config

Browse Source
This commit is contained in:
Jactus
2020-11-25 14:58:23 +08:00
parent fcbafde741
commit 3520d3b108
6 changed files with 131 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/workflow_by_code_sfm.py
View File

@@ -62,8 +62,8 @@ if __name__ == "__main__":
"kwargs": {
"d_feat": 6,
"hidden_size": 64,
"output_dim" : 1,
"freq_dim" : 15,
"output_dim": 1,
"freq_dim": 15,
"dropout_W": 0.5,
"dropout_U": 0.5,
"n_epochs": 10,
@@ -72,9 +72,9 @@ if __name__ == "__main__":
"early_stop": 20,
"eval_steps": 5,
"loss": "mse",
"lr_decay" : 0.96,
"lr_decay_steps" : 100,
"optimizer" : "gd",
"lr_decay": 0.96,
"lr_decay_steps": 100,
"optimizer": "gd",
"GPU": 1,
"seed": 0,
},