liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00
Code Activity

Update

Browse Source
This commit is contained in:
lwwang1995
2020-11-28 00:10:09 +08:00
parent d7fc90ddcd
commit 47cbfdc50c
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/benchmarks/GATs/workflow_config_gats.yaml
View File

@@ -40,19 +40,19 @@ port_analysis_config: &port_analysis_config
min_cost: 5
task:
model:
class: GAT_Classic
module_path: qlib.contrib.model.pytorch_gats_classic
class: GATs
module_path: qlib.contrib.model.pytorch_gats
kwargs:
d_feat: 6
hidden_size: 64
num_layers: 2
dropout: 0.0
dropout: 0.7
n_epochs: 200
lr: 1e-3
lr: 1e-4
early_stop: 20
metric: loss
loss: mse
base_model: GRU
base_model: LSTM
seed: 0
GPU: 0
dataset: