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

REPST #3

Merged
czzhangheng merged 42 commits from REPST into main 2025-12-20 16:03:22 +08:00
Conversation 0 Commits 42 Files Changed 465 +10747 -3684
465 changed files with 10747 additions and 3684 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

229
.vscode/launch.json vendored
View File

@ -6,233 +6,12 @@
"configurations": [
{
"name": "DDGCRN: METR-LA",
"name": "train",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"program": "train.py",
"console": "integratedTerminal",
"args": "--config ./config/DDGCRN/METR-LA.yaml"
},
// STID 模型组
{
"name": "STID: PEMS-BAY",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/PEMS-BAY.yaml"
},
{
"name": "STID: METR-LA",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/METR-LA.yaml"
},
{
"name": "STID: PEMSD4",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/PEMSD4.yaml"
},
{
"name": "STID: BJTaxi-InFlow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/BJTaxi_Inflow.yaml"
},
{
"name": "STID: BJTaxi-OutFlow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/BJTaxi_Outflow.yaml"
},
{
"name": "STID: NYCBike-InFlow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/NYCBike_Inflow.yaml"
},
{
"name": "STID: NYCBike-OutFlow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/NYCBike_Outflow.yaml"
},
{
"name": "STID: SolarEnergy",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/SolarEnergy.yaml"
},
// REPST 模型组
{
"name": "REPST: PEMSD8",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/PEMSD8.yaml"
},
{
"name": "REPST: BJTaxi-InFlow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/BJTaxi-Inflow.yaml"
},
{
"name": "REPST: NYCBike-outflow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/NYCBike-outflow.yaml"
},
{
"name": "REPST: NYCBike-inflow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/NYCBike-inflow.yaml"
},
{
"name": "REPST: PEMS-BAY",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/PEMS-BAY.yaml"
},
{
"name": "REPST: METR-LA",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/METR-LA.yaml"
},
{
"name": "REPST: SolarEnergy",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/SolarEnergy.yaml"
},
{
"name": "REPST: BeijingAirQuality",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/BeijingAirQuality.yaml"
},
{
"name": "REPST: AirQuality",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/REPST/AirQuality.yaml"
},
// AEPSA 模型组
{
"name": "AEPSA: PEMS-BAY",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/PEMS-BAY.yaml"
},
{
"name": "AEPSA: METR-LA",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/METR-LA.yaml"
},
{
"name": "AEPSA: AirQuality",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/AirQuality.yaml"
},
{
"name": "AEPSA: BJTaxi-Inflow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/BJTaxi-Inflow.yaml"
},
{
"name": "AEPSA: BJTaxi-outflow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/BJTaxi-outflow.yaml"
},
{
"name": "AEPSA: NYCBike-inflow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/NYCBike-inflow.yaml"
},
{
"name": "AEPSA: NYCBike-outflow",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/NYCBike-outflow.yaml"
},
{
"name": "AEPSA: SolarEnergy",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/SolarEnergy.yaml"
},
{
"name": "AEPSA_v2: METR-LA",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/v2_METR-LA.yaml"
},
{
"name": "AEPSA_v2: SolarEnergy",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/AEPSA/v2_SolarEnergy.yaml"
},
"justMyCode": false
}
]
}

12
config/AGCRN/AirQuality.yaml
View File

@ -13,7 +13,7 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
num_nodes: 12
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
@ -24,6 +24,7 @@ model:
embed_dim: 10
input_dim: 6
num_layers: 2
num_nodes: 12
output_dim: 6
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 1024
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 1024
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 207
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 128
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 128
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 358
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 307
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 2
num_nodes: 883
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,6 +24,7 @@ model:
embed_dim: 2
input_dim: 1
num_layers: 2
num_nodes: 170
output_dim: 1
rnn_units: 64
@ -38,13 +39,16 @@ 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:
mae_thresh: 0.0
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,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 16
column_wise: false
@ -12,17 +13,20 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
num_nodes: 12
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
@ -36,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,13 +17,16 @@ 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
@ -36,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,13 +17,16 @@ 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
@ -36,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,13 +13,14 @@ data:
input_dim: 1
lag: 12
normalizer: std
num_nodes: 13
num_nodes: 200
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
input_dim: 1
num_nodes: 200
output_dim: 1
train:
@ -39,7 +40,7 @@ train:
- 40
- 70
lr_init: 0.003
mae_thresh:
mae_thresh: null
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1

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

@ -4,6 +4,7 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 16
column_wise: false
@ -16,13 +17,16 @@ 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
@ -36,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,17 +13,20 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
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
@ -36,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,17 +13,20 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
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
@ -36,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +22,7 @@ model:
d: 1
drift: true
input_dim: 1
num_nodes: 358
output_dim: 1
p: 2
q: 1
@ -39,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +22,7 @@ model:
d: 1
drift: true
input_dim: 1
num_nodes: 307
output_dim: 1
p: 2
q: 1
@ -37,13 +38,16 @@ 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:
mae_thresh: 0.0
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,5 +51,4 @@ train:
output_dim: 1
plot: true
real_value: true
seed: 12
weight_decay: 0

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

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

2
config/ARIMA/PEMSD7.yaml
View File

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

8
config/ARIMA/PEMSD8.yaml
View File

@ -22,6 +22,7 @@ model:
d: 1
drift: true
input_dim: 1
num_nodes: 170
output_dim: 1
p: 2
q: 1
@ -37,7 +38,11 @@ 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
@ -45,5 +50,4 @@ train:
output_dim: 1
plot: false
real_value: true
seed: 10
weight_decay: 0

7
config/ARIMA/SolarEnergy.yaml
View File

@ -4,6 +4,7 @@ basic:
mode: train
model: ARIMA
seed: 2023
data:
batch_size: 16
column_wise: false
@ -16,13 +17,16 @@ 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
@ -36,11 +40,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 137
plot: false
real_value: true
seed: 10
weight_decay: 0

3
config/AEPSA/AirQuality.yaml → config/ASTRA/AirQuality.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: AirQuality
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -32,6 +32,7 @@ model:
seq_len: 24
stride: 7
word_num: 1000
output_dim: 6
train:
batch_size: 16

3
config/AEPSA/BJTaxi-Inflow.yaml → config/ASTRA/BJTaxi-InFlow.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: BJTaxi-InFlow
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -32,6 +32,7 @@ model:
seq_len: 24
stride: 7
word_num: 1000
output_dim: 1
train:
batch_size: 32

3
config/AEPSA/BJTaxi-outflow.yaml → config/ASTRA/BJTaxi-OutFlow.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: BJTaxi-OutFlow
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -17,6 +17,7 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
output_dim: 1
model:
d_ff: 128

3
config/AEPSA/METR-LA.yaml → config/ASTRA/METR-LA.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: METR-LA
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -32,6 +32,7 @@ model:
seq_len: 24
stride: 7
word_num: 1000
output_dim: 1
train:
batch_size: 16

5
config/AEPSA/NYCBike-inflow.yaml → config/ASTRA/NYCBike-InFlow.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: NYCBike-InFlow
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -14,7 +14,7 @@ data:
lag: 24
normalizer: std
num_nodes: 128
steps_per_day: 24
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
@ -32,6 +32,7 @@ model:
seq_len: 24
stride: 7
word_num: 1000
output_dim: 1
train:
batch_size: 32

5
config/AEPSA/NYCBike-outflow.yaml → config/ASTRA/NYCBike-OutFlow.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: NYCBike-OutFlow
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -14,7 +14,7 @@ data:
lag: 24
normalizer: std
num_nodes: 128
steps_per_day: 24
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
@ -32,6 +32,7 @@ model:
seq_len: 24
stride: 7
word_num: 1000
output_dim: 1
train:
batch_size: 32

3
config/AEPSA/PEMS-BAY.yaml → config/ASTRA/PEMS-BAY.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: PEMS-BAY
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -32,6 +32,7 @@ model:
seq_len: 24
stride: 7
word_num: 1000
output_dim: 1
train:
batch_size: 16

3
config/AEPSA/SolarEnergy.yaml → config/ASTRA/SolarEnergy.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: SolarEnergy
device: cuda:0
mode: train
model: AEPSA
model: ASTRA
seed: 2023
data:
@ -32,6 +32,7 @@ model:
seq_len: 24
stride: 7
word_num: 1000
output_dim: 1
train:
batch_size: 64

58
config/ASTRA_v2/AirQuality.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: AirQuality
device: cuda:0
mode: train
model: ASTRA_v2
seed: 2023
data:
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 6
n_heads: 1
num_nodes: 35
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
output_dim: 6
graph_dim: 64
graph_embed_dim: 10
train:
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 6
plot: false
weight_decay: 0

58
config/ASTRA_v2/BJTaxi-InFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: BJTaxi-InFlow
device: cuda:0
mode: train
model: ASTRA_v2
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 1024
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

58
config/ASTRA_v2/BJTaxi-OutFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: BJTaxi-OutFlow
device: cuda:0
mode: train
model: ASTRA_v2
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 1024
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

59
config/ASTRA_v2/METR-LA.yaml Normal file
View File

@ -0,0 +1,59 @@
basic:
dataset: METR-LA
device: cuda:0
mode: train
model: ASTRA_v2
seed: 2023
data:
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 207
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 207
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 1000
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
weight_decay: 0

58
config/ASTRA_v2/NYCBike-InFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: NYCBike-InFlow
device: cuda:0
mode: train
model: ASTRA_v2
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 128
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

58
config/ASTRA_v2/NYCBike-OutFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: NYCBike-OutFlow
device: cuda:0
mode: train
model: ASTRA_v2
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 128
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

15
config/REPST/BJTaxi-Inflow.yaml → config/ASTRA_v2/PEMS-BAY.yaml
View File

@ -1,8 +1,8 @@
basic:
dataset: BJTaxi-InFlow
dataset: PEMS-BAY
device: cuda:0
mode: train
model: REPST
model: ASTRA_v2
seed: 2023
data:
@ -13,12 +13,13 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
steps_per_day: 48
num_nodes: 325
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
@ -26,12 +27,15 @@ model:
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 1024
num_nodes: 325
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 16
@ -51,5 +55,4 @@ train:
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
weight_decay: 0

10
config/AEPSA/v2_SolarEnergy.yaml → config/ASTRA_v2/SolarEnergy.yaml
View File

@ -2,11 +2,11 @@ basic:
dataset: SolarEnergy
device: cuda:0
mode: train
model: AEPSA_v2
model: ASTRA_v2
seed: 2023
data:
batch_size: 64
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
@ -19,6 +19,7 @@ data:
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
@ -32,9 +33,12 @@ model:
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 64
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15

59
config/ASTRA_v3/AirQuality.yaml Normal file
View File

@ -0,0 +1,59 @@
basic:
dataset: AirQuality
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 6
n_heads: 1
num_nodes: 35
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
output_dim: 6
graph_dim: 64
graph_embed_dim: 10
output_dim: 6
train:
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 6
plot: false
weight_decay: 0

58
config/ASTRA_v3/BJTaxi-InFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: BJTaxi-InFlow
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 1024
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

58
config/ASTRA_v3/BJTaxi-OutFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: BJTaxi-OutFlow
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 1024
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

59
config/ASTRA_v3/METR-LA.yaml Normal file
View File

@ -0,0 +1,59 @@
basic:
dataset: METR-LA
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 207
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 207
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 1000
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
real_value: true
weight_decay: 0

58
config/ASTRA_v3/NYCBike-InFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: NYCBike-InFlow
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 128
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

58
config/ASTRA_v3/NYCBike-OutFlow.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: NYCBike-OutFlow
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 32
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 128
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 32
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

58
config/ASTRA_v3/PEMS-BAY.yaml Executable file
View File

@ -0,0 +1,58 @@
basic:
dataset: PEMS-BAY
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 16
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:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 325
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

58
config/ASTRA_v3/SolarEnergy.yaml Normal file
View File

@ -0,0 +1,58 @@
basic:
dataset: SolarEnergy
device: cuda:0
mode: train
model: ASTRA_v3
seed: 2023
data:
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
input_dim: 1
lag: 24
normalizer: std
num_nodes: 137
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb: 3
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
n_heads: 1
num_nodes: 137
patch_len: 6
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
graph_dim: 64
graph_embed_dim: 10
output_dim: 1
train:
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15
epochs: 100
grad_norm: false
log_step: 100
loss_func: mae
lr_decay: true
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 1
plot: false
weight_decay: 0

6
config/DCRNN/AirQuality.yaml
View File

@ -13,7 +13,7 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
num_nodes: 12
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
@ -25,6 +25,7 @@ model:
input_dim: 6
l1_decay: 0
max_diffusion_step: 2
num_nodes: 12
num_rnn_layers: 2
output_dim: 6
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 1024
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 1024
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 207
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 128
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 128
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 358
num_rnn_layers: 1
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 307
num_rnn_layers: 2
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.1
lr_decay_step: 10,20,40,80
lr_init: 0.001
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 883
num_rnn_layers: 1
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +25,7 @@ model:
input_dim: 1
l1_decay: 0
max_diffusion_step: 2
num_nodes: 170
num_rnn_layers: 1
output_dim: 1
rnn_units: 64
@ -44,11 +45,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,11 +45,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 16
column_wise: false
@ -12,19 +13,22 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
num_nodes: 12
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
@ -38,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,15 +17,18 @@ 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
@ -38,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,15 +17,18 @@ 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
@ -38,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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: 13
num_nodes: 200
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
@ -25,6 +25,7 @@ model:
horizon: 12
input_dim: 1
num_layers: 1
num_nodes: 200
output_dim: 1
rnn_units: 32
use_day: true
@ -47,7 +48,7 @@ train:
- 40
- 70
lr_init: 0.003
mae_thresh:
mae_thresh: null
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1

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

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

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

@ -4,6 +4,7 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,19 +13,22 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
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
@ -38,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,19 +13,22 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
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
@ -38,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +23,7 @@ model:
embed_dim: 12
input_dim: 1
num_layers: 1
num_nodes: 358
output_dim: 1
rnn_units: 64
use_day: true
@ -41,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +23,7 @@ model:
embed_dim: 10
input_dim: 1
num_layers: 1
num_nodes: 307
output_dim: 1
rnn_units: 64
use_day: true
@ -41,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,5 +51,4 @@ train:
output_dim: 1
plot: true
real_value: true
seed: 12
weight_decay: 0

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

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

2
config/DDGCRN/PEMSD7.yaml
View File

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

1
config/DDGCRN/PEMSD8.yaml
View File

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

7
config/DDGCRN/SolarEnergy.yaml
View File

@ -4,6 +4,7 @@ basic:
mode: train
model: DDGCRN
seed: 2023
data:
batch_size: 16
column_wise: false
@ -16,15 +17,18 @@ 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
@ -38,11 +42,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 16
column_wise: false
@ -12,10 +13,11 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
num_nodes: 12
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -29,9 +31,11 @@ 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
@ -45,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -29,9 +31,11 @@ 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
@ -45,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -29,9 +31,11 @@ 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
@ -45,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 16
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -29,9 +31,11 @@ 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
@ -45,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,10 +13,11 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -29,9 +31,11 @@ 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
@ -45,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,10 +13,11 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -29,9 +31,11 @@ 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
@ -45,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +31,7 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 358
num_nodes: 358
output_dim: 1
w_kernel: 1
window: 12
@ -48,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +31,7 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 307
num_nodes: 307
output_dim: 1
w_kernel: 1
window: 12
@ -48,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +31,7 @@ model:
n_kernels: 32
n_layers: 3
n_multiv: 883
num_nodes: 883
output_dim: 1
w_kernel: 1
window: 12
@ -48,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +31,7 @@ model:
n_kernels: 32
n_layers: 6
n_multiv: 170
num_nodes: 170
output_dim: 1
w_kernel: 1
window: 12
@ -48,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: DSANET
seed: 2023
data:
batch_size: 16
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
batch_size: 64
d_inner: 2048
@ -29,9 +31,11 @@ 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
@ -45,11 +49,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 16
column_wise: false
@ -12,10 +13,11 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
num_nodes: 12
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -24,11 +26,13 @@ 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
@ -42,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -24,11 +26,13 @@ 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
@ -42,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -24,11 +26,13 @@ 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
@ -42,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 16
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -24,11 +26,13 @@ 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
@ -42,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,10 +13,11 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -24,11 +26,13 @@ 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
@ -42,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 32
column_wise: false
@ -12,10 +13,11 @@ data:
input_dim: 1
lag: 24
normalizer: std
num_nodes: 1024
num_nodes: 128
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -24,11 +26,13 @@ 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
@ -42,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +26,7 @@ model:
in_len: 12
input_dim: 1
num_layers: 1
num_nodes: 358
output_dim: 1
rnn_units: 64
top_k: 2
@ -45,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +23,7 @@ model:
cycle_len: 288
in_len: 12
input_dim: 1
num_nodes: 307
output_dim: 1
train:
@ -38,11 +39,10 @@ train:
lr_decay_rate: 0.5
lr_decay_step: 5,20,40,65
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,6 +22,7 @@ model:
batch_size: 64
in_len: 12
input_dim: 1
num_nodes: 883
output_dim: 1
train:
@ -37,11 +38,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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,11 +43,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: 0.0
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: 716
num_nodes: 307
steps_per_day: 288
test_ratio: 0.2
val_ratio: 0.2
@ -22,6 +22,7 @@ model:
batch_size: 64
in_len: 12
input_dim: 1
num_nodes: 307
output_dim: 1
train:
@ -37,7 +38,7 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh:
mae_thresh: null
mape_thresh: 0.0
max_grad_norm: 5
output_dim: 1

7
config/EXP/SolarEnergy.yaml
View File

@ -4,6 +4,7 @@ basic:
mode: train
model: EXP
seed: 2023
data:
batch_size: 16
column_wise: false
@ -16,6 +17,7 @@ data:
steps_per_day: 24
test_ratio: 0.2
val_ratio: 0.2
model:
cheb_order: 2
embed_dim: 10
@ -24,11 +26,13 @@ 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
@ -42,11 +46,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 16
column_wise: false
@ -12,20 +13,23 @@ data:
input_dim: 6
lag: 24
normalizer: std
num_nodes: 35
num_nodes: 12
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
@ -39,11 +43,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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,6 +4,7 @@ basic:
mode: train
model: EXPB
seed: 2023
data:
batch_size: 32
column_wise: false
@ -16,16 +17,19 @@ 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
@ -39,11 +43,10 @@ train:
lr_decay_rate: 0.3
lr_decay_step: 5,20,40,70
lr_init: 0.003
mae_thresh: ''
mae_thresh: 0.0
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