liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 16:26:55 +08:00
Code Activity

Update workflow_config_sfm.yaml

Browse Source
This commit is contained in:
Haoyu Wang
2020-11-26 16:08:21 +08:00
committed by GitHub
parent 056951605b
commit f7e375ae68
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/benchmarks/SFM/workflow_config_sfm.yaml
View File

@@ -31,21 +31,21 @@ task:
kwargs: kwargs:
d_feat: 6 d_feat: 6
hidden_size: 64 hidden_size: 64
output_dim: 1 output_dim: 32
freq_dim: 20 freq_dim: 25
dropout_W: 0.5 dropout_W: 0.5
dropout_U: 0.5 dropout_U: 0.5
n_epochs: 10 n_epochs: 20
lr: 1e-3 lr: 1e-3
batch_size: 800 batch_size: 1600
early_stop: 20 early_stop: 20
eval_steps: 5 eval_steps: 5
loss: mse loss: mse
lr_decay: 0.96 lr_decay: 0.96
lr_decay_steps: 100 lr_decay_steps: 100
optimizer: gd optimizer: adam
GPU: 1 GPU: 1
seed: 0 seed: 710
dataset: dataset:
class: DatasetH class: DatasetH
module_path: qlib.data.dataset module_path: qlib.data.dataset