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

Compare commits

base: czzhangheng:5a7ec07a12b8114234fa45e2e3ec817c31e39faf
Branches Tags
czzhangheng:main
czzhangheng:REPST
czzhangheng:new-model
..
compare: czzhangheng:ce448caaeaf6e39eab976cd4ba0c812a450b7c24
Branches Tags
czzhangheng:REPST
czzhangheng:main
czzhangheng:new-model

No commits in common. "5a7ec07a12b8114234fa45e2e3ec817c31e39faf" and "ce448caaeaf6e39eab976cd4ba0c812a450b7c24" have entirely different histories.
5a7ec07a12 ... ce448caaea

467 changed files with 3777 additions and 10840 deletions
Show Stats Expand all files Collapse all files

229
.vscode/launch.json vendored
View File

@ -6,12 +6,233 @@
"configurations": [
{
"name": "train",
"name": "DDGCRN: METR-LA",
"type": "debugpy",
"request": "launch",
"program": "train.py",
"program": "run.py",
"console": "integratedTerminal",
"justMyCode": false
}
"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"
},
]
}

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

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

10
config/FPT/BJTaxi-InFlow.yaml → config/AEPSA/BJTaxi-InFlow.yaml
View File

@ -2,9 +2,8 @@ basic:
dataset: BJTaxi-InFlow
device: cuda:0
mode: train
model: FPT
model: AEPSA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,9 +16,10 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d_model: 768
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
@ -29,7 +29,7 @@ model:
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
train:
batch_size: 32
debug: false

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

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

10
config/FPT/BJTaxi-OutFlow.yaml → config/AEPSA/BJTaxi-OutFlow.yaml
View File

@ -2,9 +2,8 @@ basic:
dataset: BJTaxi-OutFlow
device: cuda:0
mode: train
model: FPT
model: AEPSA
seed: 2023
data:
batch_size: 32
column_wise: false
@ -17,9 +16,10 @@ data:
steps_per_day: 48
test_ratio: 0.2
val_ratio: 0.2
model:
d_model: 768
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
@ -29,7 +29,7 @@ model:
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
train:
batch_size: 32
debug: false

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

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

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

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

51
config/AEPSA/NYCBike-InFlow.yaml Normal file
View File

@ -0,0 +1,51 @@
basic:
dataset: NYCBike-InFlow
device: cuda:0
mode: train
model: AEPSA
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:
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
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

51
config/AEPSA/NYCBike-OutFlow.yaml Normal file
View File

@ -0,0 +1,51 @@
basic:
dataset: NYCBike-OutFlow
device: cuda:0
mode: train
model: AEPSA
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:
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
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

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

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

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

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

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

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

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

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

7
config/FPT/METR-LA.yaml → config/AEPSA/v2_METR-LA.yaml
View File

@ -2,7 +2,7 @@ basic:
dataset: METR-LA
device: cuda:0
mode: train
model: FPT
model: AEPSA_v2
seed: 2023
data:
@ -19,7 +19,9 @@ data:
val_ratio: 0.2
model:
d_model: 768
d_ff: 128
d_model: 64
dropout: 0.2
gpt_layers: 9
gpt_path: ./GPT-2
input_dim: 1
@ -29,6 +31,7 @@ model:
pred_len: 24
seq_len: 24
stride: 7
word_num: 1000
train:
batch_size: 16

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

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

14
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
weight_decay: 0
seed: 10
weight_decay: 0

12
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
weight_decay: 0
seed: 10
weight_decay: 0

12
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
weight_decay: 0
seed: 10
weight_decay: 0

12
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
weight_decay: 0
seed: 10
weight_decay: 0

12
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
weight_decay: 0
seed: 10
weight_decay: 0

12
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
weight_decay: 0
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

5
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
weight_decay: 0
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

58
config/ASTRA_v2/AirQuality.yaml
View File

@ -1,58 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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
View File

@ -1,59 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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

58
config/ASTRA_v2/PEMS-BAY.yaml
View File

@ -1,58 +0,0 @@
basic:
dataset: PEMS-BAY
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: 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

59
config/ASTRA_v3/AirQuality.yaml
View File

@ -1,59 +0,0 @@
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-OutFlow.yaml
View File

@ -1,58 +0,0 @@
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
View File

@ -1,59 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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
View File

@ -1,58 +0,0 @@
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

8
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
weight_decay: 0.0001
seed: 10
weight_decay: 0.0001

6
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
weight_decay: 0.0001
seed: 10
weight_decay: 0.0001

6
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
weight_decay: 0.0001
seed: 10
weight_decay: 0.0001

6
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
weight_decay: 0.0001
seed: 10
weight_decay: 0.0001

6
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
weight_decay: 0.0001
seed: 10
weight_decay: 0.0001

6
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
weight_decay: 0.0001
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

5
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
weight_decay: 0
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

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