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

update

Browse Source
This commit is contained in:
Alex Wang
2020-11-26 14:35:16 +08:00
parent 27f19c1f1b
commit 28b11886dd
2 changed files with 158 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
examples/workflow_by_code_sfm.py
View File

@@ -71,21 +71,22 @@ if __name__ == "__main__":
"module_path": "qlib.contrib.model.pytorch_sfm",
"kwargs": {
"d_feat": 6,
"hidden_size": 32,
"output_dim": 16,
"freq_dim": 25,
"hidden_size": 64,
"output_dim" : 32,
"freq_dim" : 25,
"dropout_W": 0.5,
"dropout_U": 0.5,
"n_epochs": 200,
"lr": 1e-3,
"batch_size": 200,
"n_epochs": 15,
"lr": 1e-2,
"metric": "",
"batch_size": 1600,
"early_stop": 20,
"eval_steps": 5,
"loss": "mse",
"lr_decay": 0.96,
"lr_decay_steps": 100,
"optimizer": "adam",
"GPU": 1,
"lr_decay" : 0.96,
"lr_decay_steps" : 100,
"optimizer" : "adam",
"GPU": 3,
"seed": 710,
},
},