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

add astra-pemsbay v2

This commit is contained in:
czzhangheng 2025-12-02 09:40:24 +08:00
parent c4414dd5d9
commit e50f443470
2 changed files with 54 additions and 264 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
config/AEPSA/v2_PEMS-BAY.yaml Executable file
View File

@ -0,0 +1,54 @@
basic:
dataset: PEMS-BAY
device: cuda:0
mode: train
model: AEPSA_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:
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
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

264
test_results.txt
View File

@ -734,270 +734,6 @@ Traceback (most recent call last):
AttributeError: 'NoneType' object has no attribute 'to'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/NYCBike-InFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD4.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/METR-LA.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/AirQuality.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/NYCBike-OutFlow.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD8.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7(L).yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD3.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/SolarEnergy.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/Hainan.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7.yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/TWDGCN/PEMSD7(M).yaml
```
Traceback (most recent call last):
File "/user/czzhangheng/code/TrafficWheel/run.py", line 66, in <module>
main()
File "/user/czzhangheng/code/TrafficWheel/run.py", line 19, in main
model = init.init_model(args)
^^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/utils/initializer.py", line 14, in init_model
model = model_selector(args).to(device)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/model_selector.py", line 38, in model_selector
return TWDGCN(model_config)
^^^^^^^^^^^^^^^^^^^^
File "/user/czzhangheng/code/TrafficWheel/model/TWDGCN/TWDGCN.py", line 84, in __init__
self.num_node = args["num_nodes"]
~~~~^^^^^^^^^^^^^
KeyError: 'num_nodes'
```
#### /user/czzhangheng/code/TrafficWheel/config/STSGCN/NYCBike-InFlow.yaml
```