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

bug fixes

This commit is contained in:
meowhe 2026-01-05 14:18:54 +08:00
parent 2bfa444f8e
commit 2c54d81a67
4 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,7 +6,7 @@ basic:
seed: 2023
data:
batch_size: 32
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
@ -31,7 +31,7 @@ model:
stride: 7
train:
batch_size: 32
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15

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

@ -6,7 +6,7 @@ basic:
seed: 2023
data:
batch_size: 32
batch_size: 16
column_wise: false
days_per_week: 7
horizon: 24
@ -31,7 +31,7 @@ model:
stride: 7
train:
batch_size: 32
batch_size: 16
debug: false
early_stop: true
early_stop_patience: 15

1
config/REPST/BJTaxi-InFlow.yaml
View File

@ -27,6 +27,7 @@ model:
input_dim: 1
n_heads: 1
num_nodes: 1024
output_dim: 1
patch_len: 6
pred_len: 24
seq_len: 24

2
train.py
View File

@ -110,7 +110,7 @@ def read_config(config_path):
if __name__ == "__main__":
# 调试用
# model_list = ["iTransformer", "PatchTST", "HI"]
model_list = ["STID"]
model_list = ["REPST"]
# model_list = ["PatchTST"]
air = ["AirQuality"]