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

Fix bugs of model.

Browse Source
This commit is contained in:
lwwang1995
2020-12-07 19:49:03 +08:00
committed by you-n-g
parent fb4a2e65cc
commit 4a748525bc
3 changed files with 20 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/benchmarks/GATs/workflow_config_gats_Alpha158.yaml
View File

@@ -51,7 +51,7 @@ task:
class: GATs
module_path: qlib.contrib.model.pytorch_gats_ts
kwargs:
d_feat: 6
d_feat: 20
hidden_size: 64
num_layers: 2
dropout: 0.7
@@ -62,7 +62,7 @@ task:
loss: mse
base_model: LSTM
with_pretrain: True
model_path: "benchmarks/LSTM/model_lstm_ts.pkl"
model_path: "benchmarks/LSTM/csi300_lstm_ts.pkl"
GPU: 0
dataset:
class: TSDatasetH

2
examples/benchmarks/SFM/workflow_config_sfm_Alpha158.yaml
View File

@@ -57,7 +57,7 @@ task:
num_layers: 2
dropout: 0.0
n_epochs: 200
lr: 1e-2
lr: 1e-1
early_stop: 10
batch_size: 800
metric: loss