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

Compare commits

base: czzhangheng:f3480fccdc7b463e9ff4415212c151a81ba99482
Branches Tags
czzhangheng:main
czzhangheng:REPST
czzhangheng:new-model
..
compare: czzhangheng:2800f66dfe62775b710266e7e7aa4f98f3fc5d2b
Branches Tags
czzhangheng:REPST
czzhangheng:main
czzhangheng:new-model

No commits in common. "f3480fccdc7b463e9ff4415212c151a81ba99482" and "2800f66dfe62775b710266e7e7aa4f98f3fc5d2b" have entirely different histories.
f3480fccdc ... 2800f66dfe

110 changed files with 366 additions and 729 deletions
Show Stats Expand all files Collapse all files

13
.vscode/launch.json vendored
View File

@ -4,21 +4,14 @@
// 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387 // 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0", "version": "0.2.0",
"configurations": [ "configurations": [
{
"name": "STID_PEMS-BAY",
"type": "debugpy",
"request": "launch",
"program": "run.py",
"console": "integratedTerminal",
"args": "--config ./config/STID/PEMS-BAY.yaml"
},
{ {
"name": "STID_PEMSD4", "name": "EXP_PEMSD8",
"type": "debugpy", "type": "debugpy",
"request": "launch", "request": "launch",
"program": "run.py", "program": "run.py",
"console": "integratedTerminal", "console": "integratedTerminal",
"args": "--config ./config/STID/PEMSD4.yaml" "args": "--config ./config/DDGCRN/PEMSD8.yaml"
}, },
{ {
"name": "REPST", "name": "REPST",

5
.vscode/settings.json vendored
View File

@ -1,5 +0,0 @@
{
"python-envs.defaultEnvManager": "ms-python.python:conda",
"python-envs.defaultPackageManager": "ms-python.python:conda",
"python-envs.pythonProjects": []
}

6
config/AGCRN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "AGCRN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/AGCRN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "AGCRN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/AGCRN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "AGCRN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/AGCRN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "AGCRN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/ARIMA/Hainan.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "Hainan"
mode: "train"
device: "cuda:0"
model: "ARIMA"
data: data:
num_nodes: 13 num_nodes: 13
lag: 12 lag: 12

6
config/ARIMA/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "ARIMA"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/ARIMA/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "ARIMA"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

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

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7(L)"
mode: "train"
device: "cuda:0"
model: "ARIMA"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

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

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7(M)"
mode: "train"
device: "cuda:0"
model: "ARIMA"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

6
config/ARIMA/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "ARIMA"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

6
config/ARIMA/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "ARIMA"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

6
config/DCRNN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "DCRNN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/DCRNN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "DCRNN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/DCRNN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "DCRNN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/DCRNN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "DCRNN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/DDGCRN/Hainan.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "Hainan"
mode: "train"
device: "cuda:0"
model: "DDGCRN"
data: data:
num_nodes: 13 num_nodes: 13
lag: 12 lag: 12

6
config/DDGCRN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "DDGCRN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/DDGCRN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "DDGCRN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

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

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7(L)"
mode: "train"
device: "cuda:0"
model: "DDGCRN"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

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

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7(M)"
mode: "train"
device: "cuda:0"
model: "DDGCRN"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

6
config/DDGCRN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "DDGCRN"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

6
config/DSANET/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "DSANET"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/DSANET/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "DSANET"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/DSANET/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "DSANET"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/DSANET/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "DSANET"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/EXP/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "EXP"
data: data:

6
config/EXP/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "EXP"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/EXP/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "EXP"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/EXP/SD.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "SD"
mode: "train"
device: "cuda:0"
model: "EXP"
data: data:
num_nodes: 716 num_nodes: 716
lag: 12 lag: 12

6
config/EXPB/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "EXPB"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/GWN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "GWN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/GWN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "GWN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/GWN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "GWN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/GWN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "GWN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/MegaCRN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "MegaCRN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/MegaCRN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "MegaCRN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/MegaCRN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "MegaCRN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/MegaCRN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "MegaCRN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/NLT/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "NLT"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/NLT/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "NLT"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/NLT/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "NLT"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/NLT/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "NLT"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/PDG2SEQ/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "PDG2SEQ"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/PDG2SEQ/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "PDG2SEQ"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/PDG2SEQ/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "PDG2SEQ"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

6
config/PDG2SEQ/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "PDG2SEQ"
data: data:
add_day_in_week: true add_day_in_week: true
add_time_in_day: true add_time_in_day: true

6
config/STAEFormer/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STAEFormer"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STAEFormer/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STAEFormer"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STAEFormer/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STAEFormer"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STAEFormer/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STAEFormer"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STAWnet/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STAWnet"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STAWnet/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STAWnet"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STAWnet/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STAWnet"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STAWnet/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STAWnet"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STFGNN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STFGNN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STFGNN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STFGNN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STFGNN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STFGNN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STFGNN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STFGNN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STGCN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STGCN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STGCN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STGCN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STGCN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STGCN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STGCN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STGCN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STGNCDE/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STGNCDE"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STGNCDE/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STGNCDE"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STGNCDE/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STGNCDE"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STGNCDE/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STGNCDE"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STGNRDE/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STGNRDE"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STGNRDE/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STGNRDE"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STGNRDE/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STGNRDE"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STGNRDE/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STGNRDE"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STGODE/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STGODE"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STGODE/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STGODE"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STGODE/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STGODE"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STGODE/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STGODE"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

66
config/STID/PEMS-BAY.yaml
View File

@ -1,66 +0,0 @@
basic:
dataset: "PEMS-BAY"
mode: "train"
device: "cuda:0"
model: "STID"
data:
num_nodes: 325
lag: 24
horizon: 24
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
input_dim: 1
output_dim: 1
batch_size: 64
model:
input_dim: 3
output_dim: 1
history: 24
horizon: 24
num_nodes: 325
input_len: 24
embed_dim: 32
output_len: 24
num_layer: 3
if_node: True
node_dim: 32
if_T_i_D: True
if_D_i_W: True
temp_dim_tid: 32
temp_dim_diw: 32
time_of_day_size: 288
day_of_week_size: 7
batch_size: 64
train:
loss_func: mae
seed: 1
batch_size: 64
epochs: 300
lr_init: 0.002
weight_decay: 0.0001
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "1,50,80"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
debug: true
output_dim: 1
mae_thresh: null
mape_thresh: 0.0
log_step: 200
plot: False

6
config/STID/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STID"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STIDGCN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STIDGCN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STIDGCN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STIDGCN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STIDGCN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STIDGCN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STIDGCN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STIDGCN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STMLP/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STMLP"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STMLP/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STMLP"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STMLP/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STMLP"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STMLP/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STMLP"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/STSGCN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "STSGCN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/STSGCN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "STSGCN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/STSGCN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "STSGCN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/STSGCN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "STSGCN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/ST_SSL/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "ST_SSL"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/ST_SSL/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "ST_SSL"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/ST_SSL/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "ST_SSL"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/ST_SSL/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "ST_SSL"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/TCN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "TCN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

6
config/TCN/PEMSD4.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD4"
mode: "train"
device: "cuda:0"
model: "TCN"
data: data:
num_nodes: 307 num_nodes: 307
lag: 12 lag: 12

6
config/TCN/PEMSD7.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD7"
mode: "train"
device: "cuda:0"
model: "TCN"
data: data:
num_nodes: 883 num_nodes: 883
lag: 12 lag: 12

6
config/TCN/PEMSD8.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD8"
mode: "train"
device: "cuda:0"
model: "TCN"
data: data:
num_nodes: 170 num_nodes: 170
lag: 12 lag: 12

6
config/TWDGCN/Hainan.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "Hainan"
mode: "train"
device: "cuda:0"
model: "TWDGCN"
data: data:
num_nodes: 13 num_nodes: 13
lag: 12 lag: 12

6
config/TWDGCN/PEMSD3.yaml
View File

@ -1,9 +1,3 @@
basic:
dataset: "PEMSD3"
mode: "train"
device: "cuda:0"
model: "TWDGCN"
data: data:
num_nodes: 358 num_nodes: 358
lag: 12 lag: 12

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