35 lines
615 B
YAML
35 lines
615 B
YAML
---
|
|
base_dir: data/model
|
|
batch_size: 64
|
|
cl_decay_steps: 2000
|
|
data_type: ALL
|
|
dropout: 0
|
|
epoch: 0
|
|
epochs: 100
|
|
filter_type: random_walk
|
|
global_step: 0
|
|
graph_pkl_filename: data/sensor_graph/adj_mx.pkl
|
|
horizon: 3
|
|
l1_decay: 0
|
|
learning_rate: 0.01
|
|
loss_func: MAE
|
|
lr_decay: 0.1
|
|
lr_decay_epoch: 20
|
|
lr_decay_interval: 10
|
|
max_diffusion_step: 2
|
|
max_grad_norm: 5
|
|
method_type: GCRNN
|
|
min_learning_rate: 2.0e-06
|
|
null_val: 0
|
|
num_rnn_layers: 2
|
|
output_dim: 1
|
|
patience: 50
|
|
rnn_units: 16
|
|
seq_len: 3
|
|
test_every_n_epochs: 10
|
|
test_ratio: 0.2
|
|
use_cpu_only: false
|
|
use_curriculum_learning: true
|
|
validation_ratio: 0.1
|
|
verbose: 0
|
|
write_db: false |