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

Merge branch 'REPST' of https://github.zhang-heng.com/czzhangheng/TrafficWheel into REPST

This commit is contained in:
czzhangheng 2026-01-06 10:23:04 +08:00
parent 521d72587f 2c54d81a67
commit 3b50931324
16 changed files with 21 additions and 20 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 seed: 2023
data: data:
batch_size: 32 batch_size: 16
column_wise: false column_wise: false
days_per_week: 7 days_per_week: 7
horizon: 24 horizon: 24
@ -31,7 +31,7 @@ model:
stride: 7 stride: 7
train: train:
batch_size: 32 batch_size: 16
debug: false debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15

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

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

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

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

2
config/STID/AirQuality.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/BJTaxi-InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/BJTaxi-OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/BJTaxi_InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/BJTaxi_OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/METR-LA.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 300 epochs: 300

2
config/STID/NYCBike-InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/NYCBike-OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/NYCBike_InFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

2
config/STID/NYCBike_OutFlow.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

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

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 300 epochs: 300

2
config/STID/SolarEnergy.yaml
View File

@ -40,7 +40,7 @@ model:
train: train:
batch_size: 64 batch_size: 64
debug: true debug: false
early_stop: true early_stop: true
early_stop_patience: 15 early_stop_patience: 15
epochs: 100 epochs: 100

8
train.py
View File

@ -90,9 +90,9 @@ def read_config(config_path):
config = yaml.safe_load(file) config = yaml.safe_load(file)
# 全局配置 # 全局配置
device = "cuda:0" # 指定设备为cuda:0 device = "cuda:1" # 指定设备为cuda:0
seed = 2023 # 随机种子 seed = 2023 # 随机种子
epochs = 10 # 训练轮数 epochs = 100 # 训练轮数
# 拷贝项 # 拷贝项
config["basic"]["seed"] = seed config["basic"]["seed"] = seed
@ -121,5 +121,5 @@ if __name__ == "__main__":
all_dataset = big_dataset + mid_dataset + regular_dataset all_dataset = big_dataset + mid_dataset + regular_dataset
dataset_list = regular_dataset dataset_list = all_dataset
main(model_list, dataset_list, debug=True) main(model_list, dataset_list, debug=False)