liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-15 08:46:56 +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 min_cost: 5
task: task:
model: model:
class: GAT_Classic class: GATs
module_path: qlib.contrib.model.pytorch_gats_classic module_path: qlib.contrib.model.pytorch_gats
kwargs: kwargs:
d_feat: 6 d_feat: 6
hidden_size: 64 hidden_size: 64
num_layers: 2 num_layers: 2
dropout: 0.0 dropout: 0.7
n_epochs: 200 n_epochs: 200
lr: 1e-3 lr: 1e-4
early_stop: 20 early_stop: 20
metric: loss metric: loss
loss: mse loss: mse
base_model: GRU base_model: LSTM
seed: 0 seed: 0
GPU: 0 GPU: 0
dataset: dataset: