czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: czzhangheng:a9313390ac3062be453f52d416c74d33ca80fff4
Branches Tags
czzhangheng:main
czzhangheng:REPST
czzhangheng:new-model
..
compare: czzhangheng:e50f44347013574f8441cf4726811e8cb3863bf0
Branches Tags
czzhangheng:REPST
czzhangheng:main
czzhangheng:new-model

No commits in common. "a9313390ac3062be453f52d416c74d33ca80fff4" and "e50f44347013574f8441cf4726811e8cb3863bf0" have entirely different histories.
a9313390ac ... e50f443470

301 changed files with 1684 additions and 1959 deletions
Show Stats Expand all files Collapse all files

4
config/AEPSA/AirQuality.yaml
View File

@ -13,7 +13,7 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
@ -26,7 +26,7 @@ model:
gpt_path: ./GPT-2
input_dim: 6
n_heads: 1
num_nodes: 12
num_nodes: 35
patch_len: 6
pred_len: 24
seq_len: 24

3
config/AEPSA/BJTaxi-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: AEPSA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d_ff: 128
d_model: 64
@ -32,7 +30,6 @@ model:
seq_len: 24
stride: 7
word_num: 1000
train:
batch_size: 32
debug: false

3
config/AEPSA/BJTaxi-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: AEPSA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d_ff: 128
d_model: 64
@ -32,7 +30,6 @@ model:
seq_len: 24
stride: 7
word_num: 1000
train:
batch_size: 32
debug: false

7
config/AEPSA/NYCBike-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: AEPSA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d_ff: 128
d_model: 64
@ -26,13 +24,12 @@ model:
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 128
num_nodes: 1024
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
train:
batch_size: 32
debug: false

7
config/AEPSA/NYCBike-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: AEPSA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d_ff: 128
d_model: 64
@ -26,13 +24,12 @@ model:
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 128
num_nodes: 1024
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
train:
batch_size: 32
debug: false

12
config/AGCRN/AirQuality.yaml
View File

@ -13,7 +13,7 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 6
num_layers: 2
num_nodes: 12
output_dim: 6
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/BJTaxi-InFlow.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 1024
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/BJTaxi-OutFlow.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 1024
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/METR-LA.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 207
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/NYCBike-InFlow.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 128
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/NYCBike-OutFlow.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 128
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/PEMSD3.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 358
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/PEMSD4.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 307
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/PEMSD7.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 883
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/AGCRN/PEMSD8.yaml
View File

@ -24,7 +24,6 @@ model:
embed_dim: 2
input_dim: 1
num_layers: 2
num_nodes: 170
output_dim: 1
rnn_units: 64
@ -39,16 +38,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 12
weight_decay: 0

3
config/AGCRN/SolarEnergy.yaml
View File

@ -46,10 +46,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/ARIMA/AirQuality.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 16
column_wise: false
@ -13,20 +12,17 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
d: 1
drift: true
input_dim: 6
num_nodes: 12
output_dim: 6
p: 2
q: 1
train:
batch_size: 16
debug: false
@ -40,10 +36,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/ARIMA/BJTaxi-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,16 +16,13 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d: 1
drift: true
input_dim: 1
num_nodes: 1024
output_dim: 1
p: 2
q: 1
train:
batch_size: 32
debug: false
@ -40,10 +36,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/ARIMA/BJTaxi-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,16 +16,13 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d: 1
drift: true
input_dim: 1
num_nodes: 1024
output_dim: 1
p: 2
q: 1
train:
batch_size: 32
debug: false
@ -40,10 +36,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

5
config/ARIMA/Hainan.yaml
View File

@ -13,14 +13,13 @@ data:
input_dim: 1
lag: 12
normalizer: std
num_nodes: 200
num_nodes: 13
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
input_dim: 1
num_nodes: 200
output_dim: 1
train:
@ -40,7 +39,7 @@ train:
- 40
- 70
lr_init: 0.003
mae_thresh: null
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1

7
config/ARIMA/METR-LA.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,16 +16,13 @@ data:
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
d: 1
drift: true
input_dim: 1
num_nodes: 207
output_dim: 1
p: 2
q: 1
train:
batch_size: 16
debug: false
@ -40,10 +36,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/ARIMA/NYCBike-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,20 +12,17 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d: 1
drift: true
input_dim: 1
num_nodes: 128
output_dim: 1
p: 2
q: 1
train:
batch_size: 32
debug: false
@ -40,10 +36,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/ARIMA/NYCBike-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,20 +12,17 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d: 1
drift: true
input_dim: 1
num_nodes: 128
output_dim: 1
p: 2
q: 1
train:
batch_size: 32
debug: false
@ -40,10 +36,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/ARIMA/PEMSD3.yaml
View File

@ -22,7 +22,6 @@ model:
d: 1
drift: true
input_dim: 1
num_nodes: 358
output_dim: 1
p: 2
q: 1
@ -40,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

10
config/ARIMA/PEMSD4.yaml
View File

@ -22,7 +22,6 @@ model:
d: 1
drift: true
input_dim: 1
num_nodes: 307
output_dim: 1
p: 2
q: 1
@ -38,16 +37,13 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

1
config/ARIMA/PEMSD7(L).yaml
View File

@ -51,4 +51,5 @@ train:
output_dim: 1
plot: true
real_value: true
seed: 12
weight_decay: 0

1
config/ARIMA/PEMSD7(M).yaml
View File

@ -51,4 +51,5 @@ train:
output_dim: 1
plot: true
real_value: true
seed: 12
weight_decay: 0

2
config/ARIMA/PEMSD7.yaml
View File

@ -22,7 +22,6 @@ model:
d: 1
drift: true
input_dim: 1
num_nodes: 883
output_dim: 1
p: 2
q: 1
@ -50,4 +49,5 @@ train:
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

8
config/ARIMA/PEMSD8.yaml
View File

@ -22,7 +22,6 @@ model:
d: 1
drift: true
input_dim: 1
num_nodes: 170
output_dim: 1
p: 2
q: 1
@ -38,11 +37,7 @@ train:
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step:
- 5
- 20
- 40
- 70
lr_decay_step: [5, 20, 40, 70]
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
@ -50,4 +45,5 @@ train:
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/ARIMA/SolarEnergy.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,16 +16,13 @@ data:
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
d: 1
drift: true
input_dim: 137
num_nodes: 137
output_dim: 137
p: 2
q: 1
train:
batch_size: 16
debug: false
@ -40,10 +36,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 137
plot: false
real_value: true
seed: 10
weight_decay: 0

6
config/DCRNN/AirQuality.yaml
View File

@ -13,7 +13,7 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
@ -25,7 +25,6 @@ model:
input_dim: 6
l1_decay: 0
max_diffusion_step: 2
num_nodes: 12
num_rnn_layers: 2
output_dim: 6
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: false
seed: 10
weight_decay: 0.0001

4
config/DCRNN/BJTaxi-InFlow.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 1024
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: false
seed: 10
weight_decay: 0.0001

4
config/DCRNN/BJTaxi-OutFlow.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 1024
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: false
seed: 10
weight_decay: 0.0001

4
config/DCRNN/METR-LA.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 207
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: false
seed: 10
weight_decay: 0.0001

4
config/DCRNN/NYCBike-InFlow.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 128
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: false
seed: 10
weight_decay: 0.0001

4
config/DCRNN/NYCBike-OutFlow.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 128
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: false
seed: 10
weight_decay: 0.0001

4
config/DCRNN/PEMSD3.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 358
num_rnn_layers: 1
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DCRNN/PEMSD4.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 307
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: false
seed: 10
weight_decay: 0.0001

4
config/DCRNN/PEMSD7.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 883
num_rnn_layers: 1
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DCRNN/PEMSD8.yaml
View File

@ -25,7 +25,6 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 170
num_rnn_layers: 1
output_dim: 1
rnn_units: 64
@ -45,10 +44,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

3
config/DCRNN/SolarEnergy.yaml
View File

@ -45,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/DDGCRN/AirQuality.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 16
column_wise: false
@ -13,22 +12,19 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 12
input_dim: 6
num_layers: 1
num_nodes: 12
output_dim: 6
rnn_units: 64
use_day: true
use_week: false
train:
batch_size: 16
debug: false
@ -42,10 +38,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/DDGCRN/BJTaxi-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,18 +16,15 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 12
input_dim: 1
num_layers: 1
num_nodes: 1024
output_dim: 1
rnn_units: 64
use_day: true
use_week: false
train:
batch_size: 32
debug: false
@ -42,10 +38,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/DDGCRN/BJTaxi-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,18 +16,15 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 12
input_dim: 1
num_layers: 1
num_nodes: 1024
output_dim: 1
rnn_units: 64
use_day: true
use_week: false
train:
batch_size: 32
debug: false
@ -42,10 +38,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

5
config/DDGCRN/Hainan.yaml
View File

@ -13,7 +13,7 @@ data:
input_dim: 1
lag: 12
normalizer: std
num_nodes: 200
num_nodes: 13
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
@ -25,7 +25,6 @@ model:
horizon: 12
input_dim: 1
num_layers: 1
num_nodes: 200
output_dim: 1
rnn_units: 32
use_day: true
@ -48,7 +47,7 @@ train:
- 40
- 70
lr_init: 0.003
mae_thresh: null
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1

1
config/DDGCRN/METR-LA.yaml
View File

@ -48,4 +48,5 @@ train:
max_grad_norm: 5
output_dim: 1
plot: false
seed: 10
weight_decay: 0

9
config/DDGCRN/NYCBike-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,22 +12,19 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 12
input_dim: 1
num_layers: 1
num_nodes: 128
output_dim: 1
rnn_units: 64
use_day: true
use_week: false
train:
batch_size: 32
debug: false
@ -42,10 +38,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/DDGCRN/NYCBike-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,22 +12,19 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 12
input_dim: 1
num_layers: 1
num_nodes: 128
output_dim: 1
rnn_units: 64
use_day: true
use_week: false
train:
batch_size: 32
debug: false
@ -42,10 +38,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DDGCRN/PEMSD3.yaml
View File

@ -23,7 +23,6 @@ model:
embed_dim: 12
input_dim: 1
num_layers: 1
num_nodes: 358
output_dim: 1
rnn_units: 64
use_day: true
@ -42,10 +41,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DDGCRN/PEMSD4.yaml
View File

@ -23,7 +23,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 307
output_dim: 1
rnn_units: 64
use_day: true
@ -42,10 +41,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

1
config/DDGCRN/PEMSD7(L).yaml
View File

@ -51,4 +51,5 @@ train:
output_dim: 1
plot: true
real_value: true
seed: 12
weight_decay: 0

1
config/DDGCRN/PEMSD7(M).yaml
View File

@ -51,4 +51,5 @@ train:
output_dim: 1
plot: true
real_value: true
seed: 12
weight_decay: 0

2
config/DDGCRN/PEMSD7.yaml
View File

@ -23,7 +23,6 @@ model:
embed_dim: 12
input_dim: 1
num_layers: 1
num_nodes: 883
output_dim: 1
rnn_units: 64
use_day: true
@ -52,4 +51,5 @@ train:
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

1
config/DDGCRN/PEMSD8.yaml
View File

@ -49,4 +49,5 @@ train:
output_dim: 1
plot: false
real_value: true
seed: 12
weight_decay: 0

7
config/DDGCRN/SolarEnergy.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,18 +16,15 @@ data:
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 12
input_dim: 137
num_layers: 1
num_nodes: 137
output_dim: 137
rnn_units: 64
use_day: true
use_week: false
train:
batch_size: 16
debug: false
@ -42,10 +38,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 137
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/DSANET/AirQuality.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 16
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -31,11 +29,9 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 35
num_nodes: 12
output_dim: 6
w_kernel: 1
window: 24
train:
batch_size: 16
debug: false
@ -49,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/DSANET/BJTaxi-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -31,11 +29,9 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 1024
num_nodes: 1024
output_dim: 1
w_kernel: 1
window: 24
train:
batch_size: 32
debug: false
@ -49,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/DSANET/BJTaxi-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -31,11 +29,9 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 1024
num_nodes: 1024
output_dim: 1
w_kernel: 1
window: 24
train:
batch_size: 32
debug: false
@ -49,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/DSANET/METR-LA.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -31,11 +29,9 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 207
num_nodes: 207
output_dim: 1
w_kernel: 1
window: 12
train:
batch_size: 16
debug: false
@ -49,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/DSANET/NYCBike-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -31,11 +29,9 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 1024
num_nodes: 128
output_dim: 1
w_kernel: 1
window: 24
train:
batch_size: 32
debug: false
@ -49,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/DSANET/NYCBike-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -31,11 +29,9 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 1024
num_nodes: 128
output_dim: 1
w_kernel: 1
window: 24
train:
batch_size: 32
debug: false
@ -49,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DSANET/PEMSD3.yaml
View File

@ -31,7 +31,6 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 358
num_nodes: 358
output_dim: 1
w_kernel: 1
window: 12
@ -49,10 +48,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DSANET/PEMSD4.yaml
View File

@ -31,7 +31,6 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 307
num_nodes: 307
output_dim: 1
w_kernel: 1
window: 12
@ -49,10 +48,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DSANET/PEMSD7.yaml
View File

@ -31,7 +31,6 @@ model:
n_kernels: 32
n_layers: 3
n_multiv: 883
num_nodes: 883
output_dim: 1
w_kernel: 1
window: 12
@ -49,10 +48,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/DSANET/PEMSD8.yaml
View File

@ -31,7 +31,6 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 170
num_nodes: 170
output_dim: 1
w_kernel: 1
window: 12
@ -49,10 +48,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/DSANET/SolarEnergy.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -31,11 +29,9 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 137
num_nodes: 137
output_dim: 137
w_kernel: 1
window: 24
train:
batch_size: 16
debug: false
@ -49,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 137
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/EXP/AirQuality.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 16
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -26,13 +24,11 @@ model:
in_len: 24
input_dim: 6
num_layers: 1
num_nodes: 12
output_dim: 6
rnn_units: 64
top_k: 2
use_day: true
use_week: true
train:
batch_size: 16
debug: false
@ -46,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/EXP/BJTaxi-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -26,13 +24,11 @@ model:
in_len: 24
input_dim: 1
num_layers: 1
num_nodes: 1024
output_dim: 1
rnn_units: 64
top_k: 2
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -46,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/EXP/BJTaxi-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -26,13 +24,11 @@ model:
in_len: 24
input_dim: 1
num_layers: 1
num_nodes: 1024
output_dim: 1
rnn_units: 64
top_k: 2
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -46,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/EXP/METR-LA.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -26,13 +24,11 @@ model:
in_len: 12
input_dim: 1
num_layers: 1
num_nodes: 207
output_dim: 1
rnn_units: 64
top_k: 2
use_day: true
use_week: true
train:
batch_size: 16
debug: false
@ -46,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/EXP/NYCBike-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -26,13 +24,11 @@ model:
in_len: 24
input_dim: 1
num_layers: 1
num_nodes: 128
output_dim: 1
rnn_units: 64
top_k: 2
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -46,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/EXP/NYCBike-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -26,13 +24,11 @@ model:
in_len: 24
input_dim: 1
num_layers: 1
num_nodes: 128
output_dim: 1
rnn_units: 64
top_k: 2
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -46,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/EXP/PEMSD3.yaml
View File

@ -26,7 +26,6 @@ model:
in_len: 12
input_dim: 1
num_layers: 1
num_nodes: 358
output_dim: 1
rnn_units: 64
top_k: 2
@ -46,10 +45,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/EXP/PEMSD4.yaml
View File

@ -23,7 +23,6 @@ model:
cycle_len: 288
in_len: 12
input_dim: 1
num_nodes: 307
output_dim: 1
train:
@ -39,10 +38,11 @@ train:
lr_decay_rate: 0.5
lr_decay_step: 5,20,40,65
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/EXP/PEMSD7.yaml
View File

@ -22,7 +22,6 @@ model:
batch_size: 64
in_len: 12
input_dim: 1
num_nodes: 883
output_dim: 1
train:
@ -38,10 +37,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

3
config/EXP/PEMSD8.yaml
View File

@ -43,10 +43,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

5
config/EXP/SD.yaml
View File

@ -13,7 +13,7 @@ data:
input_dim: 1
lag: 12
normalizer: std
num_nodes: 307
num_nodes: 716
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
@ -22,7 +22,6 @@ model:
batch_size: 64
in_len: 12
input_dim: 1
num_nodes: 307
output_dim: 1
train:
@ -38,7 +37,7 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: null
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1

7
config/EXP/SolarEnergy.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -26,13 +24,11 @@ model:
in_len: 24
input_dim: 137
num_layers: 1
num_nodes: 137
output_dim: 137
rnn_units: 64
top_k: 2
use_day: true
use_week: true
train:
batch_size: 16
debug: false
@ -46,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 137
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/EXPB/AirQuality.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 16
column_wise: false
@ -13,23 +12,20 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
input_dim: 6
num_layers: 1
num_nodes: 12
output_dim: 6
patch_size: 3
rnn_units: 64
use_day: true
use_week: true
train:
batch_size: 16
debug: false
@ -43,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/EXPB/BJTaxi-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,19 +16,16 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 1024
output_dim: 1
patch_size: 3
rnn_units: 64
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -43,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/EXPB/BJTaxi-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,19 +16,16 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 1024
output_dim: 1
patch_size: 3
rnn_units: 64
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -43,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/EXPB/METR-LA.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 64
column_wise: false
@ -17,19 +16,16 @@ data:
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 207
output_dim: 1
patch_size: 3
rnn_units: 64
use_day: true
use_week: true
train:
batch_size: 64
debug: false
@ -43,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/EXPB/NYCBike-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,23 +12,20 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 128
output_dim: 1
patch_size: 3
rnn_units: 64
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -43,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/EXPB/NYCBike-OutFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 32
column_wise: false
@ -13,23 +12,20 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 128
output_dim: 1
patch_size: 3
rnn_units: 64
use_day: true
use_week: true
train:
batch_size: 32
debug: false
@ -43,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

4
config/EXPB/PEMSD4.yaml
View File

@ -23,7 +23,6 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 307
output_dim: 1
patch_size: 3
rnn_units: 64
@ -43,10 +42,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/EXPB/SolarEnergy.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 16
column_wise: false
@ -17,19 +16,16 @@ data:
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
input_dim: 137
num_layers: 1
num_nodes: 137
output_dim: 137
patch_size: 3
rnn_units: 64
use_day: true
use_week: true
train:
batch_size: 16
debug: false
@ -43,10 +39,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 137
plot: false
real_value: true
seed: 10
weight_decay: 0

28
config/GWN/AirQuality.yaml
View File

@ -6,7 +6,7 @@ basic:
seed: 2023
data:
batch_size: 64
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
@ -20,27 +20,26 @@ data:
model:
addaptadj: true
apt_size: 10
aptinit: null
batch_size: 64
aptinit:
batch_size: 16
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
input_dim: 1
in_dim: 2
input_dim: 6
kernel_size: 2
layers: 4
num_nodes: 35
out_dim: 24
layers: 2
out_dim: 12
output_dim: 6
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 64
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
@ -52,10 +51,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

19
config/GWN/BJTaxi-InFlow.yaml
View File

@ -20,24 +20,22 @@ data:
model:
addaptadj: true
apt_size: 10
aptinit: null
batch_size: 16
aptinit:
batch_size: 32
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 4
num_nodes: 1024
out_dim: 24
layers: 2
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
@ -53,10 +51,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

18
config/GWN/BJTaxi-OutFlow.yaml
View File

@ -20,24 +20,23 @@ data:
model:
addaptadj: true
apt_size: 10
aptinit: null
aptinit:
batch_size: 32
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 4
num_nodes: 1024
out_dim: 24
layers: 2
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 32
@ -52,10 +51,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

24
config/GWN/METR-LA.yaml
View File

@ -6,7 +6,7 @@ basic:
seed: 2023
data:
batch_size: 64
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
@ -20,27 +20,26 @@ data:
model:
addaptadj: true
apt_size: 10
aptinit: null
batch_size: 64
aptinit:
batch_size: 16
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 4
num_nodes: 207
out_dim: 24
layers: 2
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 64
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
@ -52,10 +51,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

18
config/GWN/NYCBike-InFlow.yaml
View File

@ -20,24 +20,23 @@ data:
model:
addaptadj: true
apt_size: 10
aptinit: null
aptinit:
batch_size: 32
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 4
num_nodes: 128
out_dim: 24
layers: 2
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 32
@ -52,10 +51,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

24
config/GWN/NYCBike-OutFlow.yaml
View File

@ -6,7 +6,7 @@ basic:
seed: 2023
data:
batch_size: 16
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
@ -20,27 +20,26 @@ data:
model:
addaptadj: true
apt_size: 10
aptinit: null
batch_size: 16
aptinit:
batch_size: 32
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 4
num_nodes: 128
out_dim: 24
layers: 2
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 16
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
@ -52,10 +51,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

61
config/GWN/PEMS-BAY.yaml
View File

@ -1,61 +0,0 @@
basic:
dataset: PEMS-BAY
device: cuda:0
mode: train
model: GWN
seed: 2023
data:
batch_size: 64
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 325
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
addaptadj: true
apt_size: 10
aptinit: null
batch_size: 64
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
input_dim: 1
kernel_size: 2
layers: 4
num_nodes: 325
out_dim: 24
residual_channels: 32
skip_channels: 256
supports: null
train:
batch_size: 64
debug: false
early_stop: true
early_stop_patience: 15
epochs: 300
grad_norm: false
log_step: 1000
loss_func: mae
lr_decay: false
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
weight_decay: 0

18
config/GWN/PEMSD3.yaml
View File

@ -20,23 +20,30 @@ data:
model:
addaptadj: true
aptinit: null
aptinit:
batch_size: 64
blocks: 4
dilation_channels: 32
dropout: 0.3
end_channels: 512
gcn_bool: true
in_dim: 3
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 2
num_nodes: 358
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 16
@ -51,10 +58,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

18
config/GWN/PEMSD4.yaml
View File

@ -20,23 +20,30 @@ data:
model:
addaptadj: true
aptinit: null
aptinit:
batch_size: 64
blocks: 4
dilation_channels: 32
dropout: 0.3
end_channels: 512
gcn_bool: true
in_dim: 1
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 2
num_nodes: 307
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 64
@ -51,10 +58,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

18
config/GWN/PEMSD7.yaml
View File

@ -20,23 +20,30 @@ data:
model:
addaptadj: true
aptinit: null
aptinit:
batch_size: 64
blocks: 4
dilation_channels: 32
dropout: 0.3
end_channels: 512
gcn_bool: true
in_dim: 3
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 2
num_nodes: 883
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 16
@ -51,10 +58,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

18
config/GWN/PEMSD8.yaml
View File

@ -20,23 +20,30 @@ data:
model:
addaptadj: true
aptinit: null
aptinit:
batch_size: 64
blocks: 4
dilation_channels: 32
dropout: 0.3
end_channels: 512
gcn_bool: true
in_dim: 3
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 2
num_nodes: 170
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 64
@ -51,10 +58,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

19
config/GWN/SolarEnergy.yaml
View File

@ -20,24 +20,22 @@ data:
model:
addaptadj: true
apt_size: 10
aptinit: null
batch_size: 32
aptinit:
batch_size: 64
blocks: 4
dilation_channels: 32
dropout: 0.3
do_graph_conv: True
end_channels: 512
gcn_bool: true
in_dim: 1
in_dim: 2
input_dim: 1
kernel_size: 2
layers: 4
num_nodes: 137
out_dim: 24
layers: 2
out_dim: 12
output_dim: 1
residual_channels: 32
skip_channels: 256
supports: null
supports:
train:
batch_size: 64
@ -52,10 +50,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh:
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

9
config/MegaCRN/AirQuality.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: MegaCRN
seed: 2023
data:
batch_size: 16
column_wise: false
@ -13,11 +12,10 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 12
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_k: 3
cl_decay_steps: 2000
@ -25,12 +23,10 @@ model:
mem_dim: 64
mem_num: 20
num_layers: 1
num_nodes: 12
output_dim: 6
rnn_units: 64
use_curriculum_learning: true
ycov_dim: 1
train:
batch_size: 16
debug: false
@ -44,10 +40,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 6
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/MegaCRN/BJTaxi-InFlow.yaml
View File

@ -4,7 +4,6 @@ basic:
mode: train
model: MegaCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,7 +16,6 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_k: 3
cl_decay_steps: 2000
@ -25,12 +23,10 @@ model:
mem_dim: 64
mem_num: 20
num_layers: 1
num_nodes: 1024
output_dim: 1
rnn_units: 64
use_curriculum_learning: true
ycov_dim: 1
train:
batch_size: 32
debug: false
@ -44,10 +40,11 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: 0.0
mae_thresh: ''
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

Some files were not shown because too many files have changed in this diff Show More