diff --git a/config/AEPSA/AirQuality.yaml b/config/AEPSA/AirQuality.yaml index d6061d9..c2d905a 100644 --- a/config/AEPSA/AirQuality.yaml +++ b/config/AEPSA/AirQuality.yaml @@ -13,7 +13,7 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 @@ -26,7 +26,7 @@ model: gpt_path: ./GPT-2 input_dim: 6 n_heads: 1 - num_nodes: 35 + num_nodes: 12 patch_len: 6 pred_len: 24 seq_len: 24 diff --git a/config/AEPSA/BJTaxi-InFlow.yaml b/config/AEPSA/BJTaxi-InFlow.yaml index 64b53dc..a453b38 100644 --- a/config/AEPSA/BJTaxi-InFlow.yaml +++ b/config/AEPSA/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: AEPSA seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -30,6 +32,7 @@ model: seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 32 debug: false diff --git a/config/AEPSA/BJTaxi-OutFlow.yaml b/config/AEPSA/BJTaxi-OutFlow.yaml index d0cf19d..9fa0f5f 100644 --- a/config/AEPSA/BJTaxi-OutFlow.yaml +++ b/config/AEPSA/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: AEPSA seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -30,6 +32,7 @@ model: seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 32 debug: false diff --git a/config/AEPSA/NYCBike-InFlow.yaml b/config/AEPSA/NYCBike-InFlow.yaml index 2384c58..b561493 100644 --- a/config/AEPSA/NYCBike-InFlow.yaml +++ b/config/AEPSA/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: AEPSA seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -24,12 +26,13 @@ model: gpt_path: ./GPT-2 input_dim: 1 n_heads: 1 - num_nodes: 1024 + num_nodes: 128 patch_len: 6 pred_len: 24 seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 32 debug: false diff --git a/config/AEPSA/NYCBike-OutFlow.yaml b/config/AEPSA/NYCBike-OutFlow.yaml index 0b3597f..5c4da71 100644 --- a/config/AEPSA/NYCBike-OutFlow.yaml +++ b/config/AEPSA/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: AEPSA seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -24,12 +26,13 @@ model: gpt_path: ./GPT-2 input_dim: 1 n_heads: 1 - num_nodes: 1024 + num_nodes: 128 patch_len: 6 pred_len: 24 seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 32 debug: false diff --git a/config/AGCRN/AirQuality.yaml b/config/AGCRN/AirQuality.yaml index b1b904b..e400582 100644 --- a/config/AGCRN/AirQuality.yaml +++ b/config/AGCRN/AirQuality.yaml @@ -13,7 +13,7 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 6 num_layers: 2 + num_nodes: 12 output_dim: 6 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/AGCRN/BJTaxi-InFlow.yaml b/config/AGCRN/BJTaxi-InFlow.yaml index 5206e36..b01c8bf 100644 --- a/config/AGCRN/BJTaxi-InFlow.yaml +++ b/config/AGCRN/BJTaxi-InFlow.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 1024 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/AGCRN/BJTaxi-OutFlow.yaml b/config/AGCRN/BJTaxi-OutFlow.yaml index 7b5dc61..c7b687a 100644 --- a/config/AGCRN/BJTaxi-OutFlow.yaml +++ b/config/AGCRN/BJTaxi-OutFlow.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 1024 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/AGCRN/METR-LA.yaml b/config/AGCRN/METR-LA.yaml index b24e57e..20eb587 100644 --- a/config/AGCRN/METR-LA.yaml +++ b/config/AGCRN/METR-LA.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 207 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/AGCRN/NYCBike-InFlow.yaml b/config/AGCRN/NYCBike-InFlow.yaml index c1abc45..d33cab3 100644 --- a/config/AGCRN/NYCBike-InFlow.yaml +++ b/config/AGCRN/NYCBike-InFlow.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 128 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/AGCRN/NYCBike-OutFlow.yaml b/config/AGCRN/NYCBike-OutFlow.yaml index 9a5a846..1e1044f 100644 --- a/config/AGCRN/NYCBike-OutFlow.yaml +++ b/config/AGCRN/NYCBike-OutFlow.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 128 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/AGCRN/PEMSD3.yaml b/config/AGCRN/PEMSD3.yaml index aa68819..53621fd 100755 --- a/config/AGCRN/PEMSD3.yaml +++ b/config/AGCRN/PEMSD3.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 358 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/AGCRN/PEMSD4.yaml b/config/AGCRN/PEMSD4.yaml index 317b767..3aa586c 100755 --- a/config/AGCRN/PEMSD4.yaml +++ b/config/AGCRN/PEMSD4.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 307 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/AGCRN/PEMSD7.yaml b/config/AGCRN/PEMSD7.yaml index 0c0d96f..5859001 100755 --- a/config/AGCRN/PEMSD7.yaml +++ b/config/AGCRN/PEMSD7.yaml @@ -24,6 +24,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 2 + num_nodes: 883 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/AGCRN/PEMSD8.yaml b/config/AGCRN/PEMSD8.yaml index 7725af7..93a4250 100755 --- a/config/AGCRN/PEMSD8.yaml +++ b/config/AGCRN/PEMSD8.yaml @@ -24,6 +24,7 @@ model: embed_dim: 2 input_dim: 1 num_layers: 2 + num_nodes: 170 output_dim: 1 rnn_units: 64 @@ -38,13 +39,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 12 weight_decay: 0 diff --git a/config/AGCRN/SolarEnergy.yaml b/config/AGCRN/SolarEnergy.yaml index 094aec9..c6a666a 100644 --- a/config/AGCRN/SolarEnergy.yaml +++ b/config/AGCRN/SolarEnergy.yaml @@ -46,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/ARIMA/AirQuality.yaml b/config/ARIMA/AirQuality.yaml index cc2885e..45ce496 100644 --- a/config/ARIMA/AirQuality.yaml +++ b/config/ARIMA/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ARIMA seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,17 +13,20 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 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 @@ -36,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/BJTaxi-InFlow.yaml b/config/ARIMA/BJTaxi-InFlow.yaml index 0be9d12..85254a9 100644 --- a/config/ARIMA/BJTaxi-InFlow.yaml +++ b/config/ARIMA/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ARIMA seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,13 +17,16 @@ 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 @@ -36,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/BJTaxi-OutFlow.yaml b/config/ARIMA/BJTaxi-OutFlow.yaml index 14f9578..3c1d233 100644 --- a/config/ARIMA/BJTaxi-OutFlow.yaml +++ b/config/ARIMA/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ARIMA seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,13 +17,16 @@ 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 @@ -36,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/Hainan.yaml b/config/ARIMA/Hainan.yaml index 90cfda2..ed1a12a 100755 --- a/config/ARIMA/Hainan.yaml +++ b/config/ARIMA/Hainan.yaml @@ -13,13 +13,14 @@ data: input_dim: 1 lag: 12 normalizer: std - num_nodes: 13 + num_nodes: 200 steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 model: input_dim: 1 + num_nodes: 200 output_dim: 1 train: @@ -39,7 +40,7 @@ train: - 40 - 70 lr_init: 0.003 - mae_thresh: + mae_thresh: null mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 diff --git a/config/ARIMA/METR-LA.yaml b/config/ARIMA/METR-LA.yaml index 2b8598a..084e20e 100644 --- a/config/ARIMA/METR-LA.yaml +++ b/config/ARIMA/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ARIMA seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,13 +17,16 @@ 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 @@ -36,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/NYCBike-InFlow.yaml b/config/ARIMA/NYCBike-InFlow.yaml index 127493b..0da5634 100644 --- a/config/ARIMA/NYCBike-InFlow.yaml +++ b/config/ARIMA/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ARIMA seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,17 +13,20 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 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 @@ -36,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/NYCBike-OutFlow.yaml b/config/ARIMA/NYCBike-OutFlow.yaml index a1e3819..ddb85a2 100644 --- a/config/ARIMA/NYCBike-OutFlow.yaml +++ b/config/ARIMA/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ARIMA seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,17 +13,20 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 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 @@ -36,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/PEMSD3.yaml b/config/ARIMA/PEMSD3.yaml index 37bae7a..27b8605 100755 --- a/config/ARIMA/PEMSD3.yaml +++ b/config/ARIMA/PEMSD3.yaml @@ -22,6 +22,7 @@ model: d: 1 drift: true input_dim: 1 + num_nodes: 358 output_dim: 1 p: 2 q: 1 @@ -39,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/PEMSD4.yaml b/config/ARIMA/PEMSD4.yaml index 714973b..b4f0439 100755 --- a/config/ARIMA/PEMSD4.yaml +++ b/config/ARIMA/PEMSD4.yaml @@ -22,6 +22,7 @@ model: d: 1 drift: true input_dim: 1 + num_nodes: 307 output_dim: 1 p: 2 q: 1 @@ -37,13 +38,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/PEMSD7(L).yaml b/config/ARIMA/PEMSD7(L).yaml index 3eedd0d..25653c3 100755 --- a/config/ARIMA/PEMSD7(L).yaml +++ b/config/ARIMA/PEMSD7(L).yaml @@ -51,5 +51,4 @@ train: output_dim: 1 plot: true real_value: true - seed: 12 weight_decay: 0 diff --git a/config/ARIMA/PEMSD7(M).yaml b/config/ARIMA/PEMSD7(M).yaml index 5992f68..24ef88e 100755 --- a/config/ARIMA/PEMSD7(M).yaml +++ b/config/ARIMA/PEMSD7(M).yaml @@ -51,5 +51,4 @@ train: output_dim: 1 plot: true real_value: true - seed: 12 weight_decay: 0 diff --git a/config/ARIMA/PEMSD7.yaml b/config/ARIMA/PEMSD7.yaml index 6c027c6..f5a5255 100755 --- a/config/ARIMA/PEMSD7.yaml +++ b/config/ARIMA/PEMSD7.yaml @@ -22,6 +22,7 @@ model: d: 1 drift: true input_dim: 1 + num_nodes: 883 output_dim: 1 p: 2 q: 1 @@ -49,5 +50,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/PEMSD8.yaml b/config/ARIMA/PEMSD8.yaml index cbe3ed3..16e0339 100755 --- a/config/ARIMA/PEMSD8.yaml +++ b/config/ARIMA/PEMSD8.yaml @@ -22,6 +22,7 @@ model: d: 1 drift: true input_dim: 1 + num_nodes: 170 output_dim: 1 p: 2 q: 1 @@ -37,7 +38,11 @@ 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 @@ -45,5 +50,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ARIMA/SolarEnergy.yaml b/config/ARIMA/SolarEnergy.yaml index f1aeb63..9e921f0 100644 --- a/config/ARIMA/SolarEnergy.yaml +++ b/config/ARIMA/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ARIMA seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,13 +17,16 @@ 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 @@ -36,11 +40,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DCRNN/AirQuality.yaml b/config/DCRNN/AirQuality.yaml index 4922365..387b89b 100644 --- a/config/DCRNN/AirQuality.yaml +++ b/config/DCRNN/AirQuality.yaml @@ -13,7 +13,7 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 @@ -25,6 +25,7 @@ model: input_dim: 6 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 12 num_rnn_layers: 2 output_dim: 6 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.1 lr_decay_step: 10,20,40,80 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: false - seed: 10 - weight_decay: 0.0001 \ No newline at end of file + weight_decay: 0.0001 diff --git a/config/DCRNN/BJTaxi-InFlow.yaml b/config/DCRNN/BJTaxi-InFlow.yaml index b81dc86..16a3f91 100644 --- a/config/DCRNN/BJTaxi-InFlow.yaml +++ b/config/DCRNN/BJTaxi-InFlow.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 1024 num_rnn_layers: 2 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.1 lr_decay_step: 10,20,40,80 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: false - seed: 10 - weight_decay: 0.0001 \ No newline at end of file + weight_decay: 0.0001 diff --git a/config/DCRNN/BJTaxi-OutFlow.yaml b/config/DCRNN/BJTaxi-OutFlow.yaml index dfffb51..339e1ec 100644 --- a/config/DCRNN/BJTaxi-OutFlow.yaml +++ b/config/DCRNN/BJTaxi-OutFlow.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 1024 num_rnn_layers: 2 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.1 lr_decay_step: 10,20,40,80 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: false - seed: 10 - weight_decay: 0.0001 \ No newline at end of file + weight_decay: 0.0001 diff --git a/config/DCRNN/METR-LA.yaml b/config/DCRNN/METR-LA.yaml index 18fb223..15fcb44 100644 --- a/config/DCRNN/METR-LA.yaml +++ b/config/DCRNN/METR-LA.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 207 num_rnn_layers: 2 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.1 lr_decay_step: 10,20,40,80 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: false - seed: 10 - weight_decay: 0.0001 \ No newline at end of file + weight_decay: 0.0001 diff --git a/config/DCRNN/NYCBike-InFlow.yaml b/config/DCRNN/NYCBike-InFlow.yaml index bf7d773..e53a839 100644 --- a/config/DCRNN/NYCBike-InFlow.yaml +++ b/config/DCRNN/NYCBike-InFlow.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 128 num_rnn_layers: 2 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.1 lr_decay_step: 10,20,40,80 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: false - seed: 10 - weight_decay: 0.0001 \ No newline at end of file + weight_decay: 0.0001 diff --git a/config/DCRNN/NYCBike-OutFlow.yaml b/config/DCRNN/NYCBike-OutFlow.yaml index 7472459..a9ba532 100644 --- a/config/DCRNN/NYCBike-OutFlow.yaml +++ b/config/DCRNN/NYCBike-OutFlow.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 128 num_rnn_layers: 2 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.1 lr_decay_step: 10,20,40,80 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: false - seed: 10 - weight_decay: 0.0001 \ No newline at end of file + weight_decay: 0.0001 diff --git a/config/DCRNN/PEMSD3.yaml b/config/DCRNN/PEMSD3.yaml index 75f7dde..7d0e4a8 100755 --- a/config/DCRNN/PEMSD3.yaml +++ b/config/DCRNN/PEMSD3.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 358 num_rnn_layers: 1 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DCRNN/PEMSD4.yaml b/config/DCRNN/PEMSD4.yaml index 803d032..ddf3156 100755 --- a/config/DCRNN/PEMSD4.yaml +++ b/config/DCRNN/PEMSD4.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 307 num_rnn_layers: 2 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.1 lr_decay_step: 10,20,40,80 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: false - seed: 10 weight_decay: 0.0001 diff --git a/config/DCRNN/PEMSD7.yaml b/config/DCRNN/PEMSD7.yaml index e940611..8b8e43b 100755 --- a/config/DCRNN/PEMSD7.yaml +++ b/config/DCRNN/PEMSD7.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 883 num_rnn_layers: 1 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DCRNN/PEMSD8.yaml b/config/DCRNN/PEMSD8.yaml index cde60d3..709b392 100755 --- a/config/DCRNN/PEMSD8.yaml +++ b/config/DCRNN/PEMSD8.yaml @@ -25,6 +25,7 @@ model: input_dim: 1 l1_decay: 0 max_diffusion_step: 2 + num_nodes: 170 num_rnn_layers: 1 output_dim: 1 rnn_units: 64 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DCRNN/SolarEnergy.yaml b/config/DCRNN/SolarEnergy.yaml index 3bc9fc2..434abdd 100644 --- a/config/DCRNN/SolarEnergy.yaml +++ b/config/DCRNN/SolarEnergy.yaml @@ -45,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/DDGCRN/AirQuality.yaml b/config/DDGCRN/AirQuality.yaml index 954728b..7090253 100644 --- a/config/DDGCRN/AirQuality.yaml +++ b/config/DDGCRN/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DDGCRN seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,19 +13,22 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 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 @@ -38,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/BJTaxi-InFlow.yaml b/config/DDGCRN/BJTaxi-InFlow.yaml index ebd58a2..12dffa4 100644 --- a/config/DDGCRN/BJTaxi-InFlow.yaml +++ b/config/DDGCRN/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DDGCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,15 +17,18 @@ 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 @@ -38,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/BJTaxi-OutFlow.yaml b/config/DDGCRN/BJTaxi-OutFlow.yaml index 89a64b6..eb88c12 100644 --- a/config/DDGCRN/BJTaxi-OutFlow.yaml +++ b/config/DDGCRN/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DDGCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,15 +17,18 @@ 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 @@ -38,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/Hainan.yaml b/config/DDGCRN/Hainan.yaml index e22e71f..c02cd76 100755 --- a/config/DDGCRN/Hainan.yaml +++ b/config/DDGCRN/Hainan.yaml @@ -13,7 +13,7 @@ data: input_dim: 1 lag: 12 normalizer: std - num_nodes: 13 + num_nodes: 200 steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 @@ -25,6 +25,7 @@ model: horizon: 12 input_dim: 1 num_layers: 1 + num_nodes: 200 output_dim: 1 rnn_units: 32 use_day: true @@ -47,7 +48,7 @@ train: - 40 - 70 lr_init: 0.003 - mae_thresh: + mae_thresh: null mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 diff --git a/config/DDGCRN/METR-LA.yaml b/config/DDGCRN/METR-LA.yaml index 013fb5e..c18ffb7 100755 --- a/config/DDGCRN/METR-LA.yaml +++ b/config/DDGCRN/METR-LA.yaml @@ -48,5 +48,4 @@ train: max_grad_norm: 5 output_dim: 1 plot: false - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/NYCBike-InFlow.yaml b/config/DDGCRN/NYCBike-InFlow.yaml index 30846fb..1e9f2fb 100644 --- a/config/DDGCRN/NYCBike-InFlow.yaml +++ b/config/DDGCRN/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DDGCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,19 +13,22 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 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 @@ -38,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/NYCBike-OutFlow.yaml b/config/DDGCRN/NYCBike-OutFlow.yaml index b48986f..227d00e 100644 --- a/config/DDGCRN/NYCBike-OutFlow.yaml +++ b/config/DDGCRN/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DDGCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,19 +13,22 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 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 @@ -38,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/PEMSD3.yaml b/config/DDGCRN/PEMSD3.yaml index 98bebd0..3064f54 100755 --- a/config/DDGCRN/PEMSD3.yaml +++ b/config/DDGCRN/PEMSD3.yaml @@ -23,6 +23,7 @@ model: embed_dim: 12 input_dim: 1 num_layers: 1 + num_nodes: 358 output_dim: 1 rnn_units: 64 use_day: true @@ -41,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/PEMSD4.yaml b/config/DDGCRN/PEMSD4.yaml index fcf818e..b6c4073 100755 --- a/config/DDGCRN/PEMSD4.yaml +++ b/config/DDGCRN/PEMSD4.yaml @@ -23,6 +23,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 1 + num_nodes: 307 output_dim: 1 rnn_units: 64 use_day: true @@ -41,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/PEMSD7(L).yaml b/config/DDGCRN/PEMSD7(L).yaml index f9063ef..2ddc530 100755 --- a/config/DDGCRN/PEMSD7(L).yaml +++ b/config/DDGCRN/PEMSD7(L).yaml @@ -51,5 +51,4 @@ train: output_dim: 1 plot: true real_value: true - seed: 12 weight_decay: 0 diff --git a/config/DDGCRN/PEMSD7(M).yaml b/config/DDGCRN/PEMSD7(M).yaml index e7d87c3..a907f41 100755 --- a/config/DDGCRN/PEMSD7(M).yaml +++ b/config/DDGCRN/PEMSD7(M).yaml @@ -51,5 +51,4 @@ train: output_dim: 1 plot: true real_value: true - seed: 12 weight_decay: 0 diff --git a/config/DDGCRN/PEMSD7.yaml b/config/DDGCRN/PEMSD7.yaml index ef828cb..48c2129 100755 --- a/config/DDGCRN/PEMSD7.yaml +++ b/config/DDGCRN/PEMSD7.yaml @@ -23,6 +23,7 @@ model: embed_dim: 12 input_dim: 1 num_layers: 1 + num_nodes: 883 output_dim: 1 rnn_units: 64 use_day: true @@ -51,5 +52,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DDGCRN/PEMSD8.yaml b/config/DDGCRN/PEMSD8.yaml index d467cf8..05b469e 100755 --- a/config/DDGCRN/PEMSD8.yaml +++ b/config/DDGCRN/PEMSD8.yaml @@ -49,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 12 weight_decay: 0 diff --git a/config/DDGCRN/SolarEnergy.yaml b/config/DDGCRN/SolarEnergy.yaml index b23ea03..902aa98 100644 --- a/config/DDGCRN/SolarEnergy.yaml +++ b/config/DDGCRN/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DDGCRN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,15 +17,18 @@ 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 @@ -38,11 +42,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/AirQuality.yaml b/config/DSANET/AirQuality.yaml index 2147269..f5d0b7d 100644 --- a/config/DSANET/AirQuality.yaml +++ b/config/DSANET/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DSANET seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 d_inner: 2048 @@ -29,9 +31,11 @@ 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 @@ -45,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/BJTaxi-InFlow.yaml b/config/DSANET/BJTaxi-InFlow.yaml index 7d40eff..2f81ff6 100644 --- a/config/DSANET/BJTaxi-InFlow.yaml +++ b/config/DSANET/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DSANET seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 d_inner: 2048 @@ -29,9 +31,11 @@ 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 @@ -45,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/BJTaxi-OutFlow.yaml b/config/DSANET/BJTaxi-OutFlow.yaml index 38e1e4e..dc5c1bc 100644 --- a/config/DSANET/BJTaxi-OutFlow.yaml +++ b/config/DSANET/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DSANET seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 d_inner: 2048 @@ -29,9 +31,11 @@ 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 @@ -45,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/METR-LA.yaml b/config/DSANET/METR-LA.yaml index 108931b..6e920c5 100644 --- a/config/DSANET/METR-LA.yaml +++ b/config/DSANET/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DSANET seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 d_inner: 2048 @@ -29,9 +31,11 @@ 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 @@ -45,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/NYCBike-InFlow.yaml b/config/DSANET/NYCBike-InFlow.yaml index 2534078..f3cc3f8 100644 --- a/config/DSANET/NYCBike-InFlow.yaml +++ b/config/DSANET/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DSANET seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 d_inner: 2048 @@ -29,9 +31,11 @@ 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 @@ -45,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/NYCBike-OutFlow.yaml b/config/DSANET/NYCBike-OutFlow.yaml index 3131ccc..eb6c116 100644 --- a/config/DSANET/NYCBike-OutFlow.yaml +++ b/config/DSANET/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DSANET seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 d_inner: 2048 @@ -29,9 +31,11 @@ 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 @@ -45,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/PEMSD3.yaml b/config/DSANET/PEMSD3.yaml index 38dccec..c9cb07e 100755 --- a/config/DSANET/PEMSD3.yaml +++ b/config/DSANET/PEMSD3.yaml @@ -31,6 +31,7 @@ model: n_kernels: 32 n_layers: 6 n_multiv: 358 + num_nodes: 358 output_dim: 1 w_kernel: 1 window: 12 @@ -48,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/PEMSD4.yaml b/config/DSANET/PEMSD4.yaml index bba0aa4..6676526 100755 --- a/config/DSANET/PEMSD4.yaml +++ b/config/DSANET/PEMSD4.yaml @@ -31,6 +31,7 @@ model: n_kernels: 32 n_layers: 6 n_multiv: 307 + num_nodes: 307 output_dim: 1 w_kernel: 1 window: 12 @@ -48,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/PEMSD7.yaml b/config/DSANET/PEMSD7.yaml index a04d2de..8d51681 100755 --- a/config/DSANET/PEMSD7.yaml +++ b/config/DSANET/PEMSD7.yaml @@ -31,6 +31,7 @@ model: n_kernels: 32 n_layers: 3 n_multiv: 883 + num_nodes: 883 output_dim: 1 w_kernel: 1 window: 12 @@ -48,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/PEMSD8.yaml b/config/DSANET/PEMSD8.yaml index 02a46bd..5f5ce7e 100755 --- a/config/DSANET/PEMSD8.yaml +++ b/config/DSANET/PEMSD8.yaml @@ -31,6 +31,7 @@ model: n_kernels: 32 n_layers: 6 n_multiv: 170 + num_nodes: 170 output_dim: 1 w_kernel: 1 window: 12 @@ -48,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/DSANET/SolarEnergy.yaml b/config/DSANET/SolarEnergy.yaml index c2dd6eb..cc44e42 100644 --- a/config/DSANET/SolarEnergy.yaml +++ b/config/DSANET/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: DSANET seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 d_inner: 2048 @@ -29,9 +31,11 @@ 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 @@ -45,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/AirQuality.yaml b/config/EXP/AirQuality.yaml index ff7cf8d..f8dbaba 100644 --- a/config/EXP/AirQuality.yaml +++ b/config/EXP/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXP seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 @@ -24,11 +26,13 @@ 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 @@ -42,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/BJTaxi-InFlow.yaml b/config/EXP/BJTaxi-InFlow.yaml index c924453..01f1e63 100644 --- a/config/EXP/BJTaxi-InFlow.yaml +++ b/config/EXP/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXP seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 @@ -24,11 +26,13 @@ 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 @@ -42,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/BJTaxi-OutFlow.yaml b/config/EXP/BJTaxi-OutFlow.yaml index 6377e0b..4acad90 100644 --- a/config/EXP/BJTaxi-OutFlow.yaml +++ b/config/EXP/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXP seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 @@ -24,11 +26,13 @@ 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 @@ -42,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/METR-LA.yaml b/config/EXP/METR-LA.yaml index 28ef4c1..e0ecd55 100644 --- a/config/EXP/METR-LA.yaml +++ b/config/EXP/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXP seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 @@ -24,11 +26,13 @@ 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 @@ -42,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/NYCBike-InFlow.yaml b/config/EXP/NYCBike-InFlow.yaml index 34876bc..3210b53 100644 --- a/config/EXP/NYCBike-InFlow.yaml +++ b/config/EXP/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXP seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 @@ -24,11 +26,13 @@ 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 @@ -42,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/NYCBike-OutFlow.yaml b/config/EXP/NYCBike-OutFlow.yaml index 79e06ac..28e74d2 100644 --- a/config/EXP/NYCBike-OutFlow.yaml +++ b/config/EXP/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXP seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 @@ -24,11 +26,13 @@ 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 @@ -42,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/PEMSD3.yaml b/config/EXP/PEMSD3.yaml index 7e00b5f..def7295 100755 --- a/config/EXP/PEMSD3.yaml +++ b/config/EXP/PEMSD3.yaml @@ -26,6 +26,7 @@ model: in_len: 12 input_dim: 1 num_layers: 1 + num_nodes: 358 output_dim: 1 rnn_units: 64 top_k: 2 @@ -45,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/PEMSD4.yaml b/config/EXP/PEMSD4.yaml index 560d6da..cdb81aa 100755 --- a/config/EXP/PEMSD4.yaml +++ b/config/EXP/PEMSD4.yaml @@ -23,6 +23,7 @@ model: cycle_len: 288 in_len: 12 input_dim: 1 + num_nodes: 307 output_dim: 1 train: @@ -38,11 +39,10 @@ train: lr_decay_rate: 0.5 lr_decay_step: 5,20,40,65 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/PEMSD7.yaml b/config/EXP/PEMSD7.yaml index 029e356..9233f08 100755 --- a/config/EXP/PEMSD7.yaml +++ b/config/EXP/PEMSD7.yaml @@ -22,6 +22,7 @@ model: batch_size: 64 in_len: 12 input_dim: 1 + num_nodes: 883 output_dim: 1 train: @@ -37,11 +38,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/PEMSD8.yaml b/config/EXP/PEMSD8.yaml index 5061050..e8af5dc 100755 --- a/config/EXP/PEMSD8.yaml +++ b/config/EXP/PEMSD8.yaml @@ -43,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXP/SD.yaml b/config/EXP/SD.yaml index f61120b..493b443 100755 --- a/config/EXP/SD.yaml +++ b/config/EXP/SD.yaml @@ -13,7 +13,7 @@ data: input_dim: 1 lag: 12 normalizer: std - num_nodes: 716 + num_nodes: 307 steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 @@ -22,6 +22,7 @@ model: batch_size: 64 in_len: 12 input_dim: 1 + num_nodes: 307 output_dim: 1 train: @@ -37,7 +38,7 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: null mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 diff --git a/config/EXP/SolarEnergy.yaml b/config/EXP/SolarEnergy.yaml index 79e1496..de5edfe 100644 --- a/config/EXP/SolarEnergy.yaml +++ b/config/EXP/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXP seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 @@ -24,11 +26,13 @@ model: in_len: 24 input_dim: 137 num_layers: 1 + num_nodes: 137 output_dim: 137 rnn_units: 64 top_k: 2 use_day: true use_week: true + train: batch_size: 16 debug: false @@ -42,11 +46,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/AirQuality.yaml b/config/EXPB/AirQuality.yaml index 238f7f7..4b8082c 100644 --- a/config/EXPB/AirQuality.yaml +++ b/config/EXPB/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXPB seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,20 +13,23 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 input_dim: 6 num_layers: 1 + num_nodes: 12 output_dim: 6 patch_size: 3 rnn_units: 64 use_day: true use_week: true + train: batch_size: 16 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/BJTaxi-InFlow.yaml b/config/EXPB/BJTaxi-InFlow.yaml index 1eb34f4..0fb7614 100644 --- a/config/EXPB/BJTaxi-InFlow.yaml +++ b/config/EXPB/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXPB seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 input_dim: 1 num_layers: 1 + num_nodes: 1024 output_dim: 1 patch_size: 3 rnn_units: 64 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/BJTaxi-OutFlow.yaml b/config/EXPB/BJTaxi-OutFlow.yaml index b913f8f..798f5ca 100644 --- a/config/EXPB/BJTaxi-OutFlow.yaml +++ b/config/EXPB/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXPB seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 input_dim: 1 num_layers: 1 + num_nodes: 1024 output_dim: 1 patch_size: 3 rnn_units: 64 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/METR-LA.yaml b/config/EXPB/METR-LA.yaml index 3416970..343b252 100644 --- a/config/EXPB/METR-LA.yaml +++ b/config/EXPB/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXPB seed: 2023 + data: batch_size: 64 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 input_dim: 1 num_layers: 1 + num_nodes: 207 output_dim: 1 patch_size: 3 rnn_units: 64 use_day: true use_week: true + train: batch_size: 64 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/NYCBike-InFlow.yaml b/config/EXPB/NYCBike-InFlow.yaml index 2642db3..ac7b0d9 100644 --- a/config/EXPB/NYCBike-InFlow.yaml +++ b/config/EXPB/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXPB seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,20 +13,23 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 input_dim: 1 num_layers: 1 + num_nodes: 128 output_dim: 1 patch_size: 3 rnn_units: 64 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/NYCBike-OutFlow.yaml b/config/EXPB/NYCBike-OutFlow.yaml index 3501ece..a4d845e 100644 --- a/config/EXPB/NYCBike-OutFlow.yaml +++ b/config/EXPB/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXPB seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,20 +13,23 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 input_dim: 1 num_layers: 1 + num_nodes: 128 output_dim: 1 patch_size: 3 rnn_units: 64 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/PEMSD4.yaml b/config/EXPB/PEMSD4.yaml index 4e2c908..cf301c2 100755 --- a/config/EXPB/PEMSD4.yaml +++ b/config/EXPB/PEMSD4.yaml @@ -23,6 +23,7 @@ model: embed_dim: 10 input_dim: 1 num_layers: 1 + num_nodes: 307 output_dim: 1 patch_size: 3 rnn_units: 64 @@ -42,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/EXPB/SolarEnergy.yaml b/config/EXPB/SolarEnergy.yaml index 8e1a595..2d1f64e 100644 --- a/config/EXPB/SolarEnergy.yaml +++ b/config/EXPB/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: EXPB seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 10 input_dim: 137 num_layers: 1 + num_nodes: 137 output_dim: 137 patch_size: 3 rnn_units: 64 use_day: true use_week: true + train: batch_size: 16 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/GWN/AirQuality.yaml b/config/GWN/AirQuality.yaml index e1d8f4b..786219f 100644 --- a/config/GWN/AirQuality.yaml +++ b/config/GWN/AirQuality.yaml @@ -13,14 +13,14 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 model: addaptadj: true - aptinit: + aptinit: null batch_size: 16 blocks: 4 dilation_channels: 32 @@ -31,12 +31,12 @@ model: input_dim: 6 kernel_size: 2 layers: 2 + num_nodes: 12 out_dim: 12 output_dim: 6 residual_channels: 32 skip_channels: 256 - supports: - + supports: null train: batch_size: 16 @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/GWN/BJTaxi-InFlow.yaml b/config/GWN/BJTaxi-InFlow.yaml index 54a5631..f2f10c8 100644 --- a/config/GWN/BJTaxi-InFlow.yaml +++ b/config/GWN/BJTaxi-InFlow.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 32 blocks: 4 dilation_channels: 32 @@ -31,12 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 1024 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - + supports: null train: batch_size: 32 @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/GWN/BJTaxi-OutFlow.yaml b/config/GWN/BJTaxi-OutFlow.yaml index ea133e8..cef9af4 100644 --- a/config/GWN/BJTaxi-OutFlow.yaml +++ b/config/GWN/BJTaxi-OutFlow.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 32 blocks: 4 dilation_channels: 32 @@ -31,12 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 1024 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - + supports: null train: batch_size: 32 @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/GWN/METR-LA.yaml b/config/GWN/METR-LA.yaml index 96faa45..9ffb5d1 100644 --- a/config/GWN/METR-LA.yaml +++ b/config/GWN/METR-LA.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 16 blocks: 4 dilation_channels: 32 @@ -31,12 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 207 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - + supports: null train: batch_size: 16 @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/GWN/NYCBike-InFlow.yaml b/config/GWN/NYCBike-InFlow.yaml index 1f4c646..c536802 100644 --- a/config/GWN/NYCBike-InFlow.yaml +++ b/config/GWN/NYCBike-InFlow.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 32 blocks: 4 dilation_channels: 32 @@ -31,12 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 128 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - + supports: null train: batch_size: 32 @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/GWN/NYCBike-OutFlow.yaml b/config/GWN/NYCBike-OutFlow.yaml index a73d3fc..c67790b 100644 --- a/config/GWN/NYCBike-OutFlow.yaml +++ b/config/GWN/NYCBike-OutFlow.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 32 blocks: 4 dilation_channels: 32 @@ -31,12 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 128 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - + supports: null train: batch_size: 32 @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/GWN/PEMSD3.yaml b/config/GWN/PEMSD3.yaml index f3d78ac..9e75da7 100755 --- a/config/GWN/PEMSD3.yaml +++ b/config/GWN/PEMSD3.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 64 blocks: 4 dilation_channels: 32 @@ -31,19 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 358 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - - - - - - - - + supports: null train: batch_size: 16 @@ -58,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/GWN/PEMSD4.yaml b/config/GWN/PEMSD4.yaml index ceccee3..5435727 100755 --- a/config/GWN/PEMSD4.yaml +++ b/config/GWN/PEMSD4.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 64 blocks: 4 dilation_channels: 32 @@ -31,19 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 307 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - - - - - - - - + supports: null train: batch_size: 64 @@ -58,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/GWN/PEMSD7.yaml b/config/GWN/PEMSD7.yaml index 2cbfc62..7330998 100755 --- a/config/GWN/PEMSD7.yaml +++ b/config/GWN/PEMSD7.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 64 blocks: 4 dilation_channels: 32 @@ -31,19 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 883 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - - - - - - - - + supports: null train: batch_size: 16 @@ -58,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/GWN/PEMSD8.yaml b/config/GWN/PEMSD8.yaml index 88a5090..cebe500 100755 --- a/config/GWN/PEMSD8.yaml +++ b/config/GWN/PEMSD8.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 64 blocks: 4 dilation_channels: 32 @@ -31,19 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 170 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: - - - - - - - - + supports: null train: batch_size: 64 @@ -58,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/GWN/SolarEnergy.yaml b/config/GWN/SolarEnergy.yaml index 76110e1..afdce7a 100644 --- a/config/GWN/SolarEnergy.yaml +++ b/config/GWN/SolarEnergy.yaml @@ -20,7 +20,7 @@ data: model: addaptadj: true - aptinit: + aptinit: null batch_size: 64 blocks: 4 dilation_channels: 32 @@ -31,11 +31,12 @@ model: input_dim: 1 kernel_size: 2 layers: 2 + num_nodes: 137 out_dim: 12 output_dim: 1 residual_channels: 32 skip_channels: 256 - supports: + supports: null train: batch_size: 64 @@ -50,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/MegaCRN/AirQuality.yaml b/config/MegaCRN/AirQuality.yaml index 66583fe..c7fdfe8 100644 --- a/config/MegaCRN/AirQuality.yaml +++ b/config/MegaCRN/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: MegaCRN seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 3 cl_decay_steps: 2000 @@ -23,10 +25,12 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 12 output_dim: 6 rnn_units: 64 use_curriculum_learning: true ycov_dim: 1 + train: batch_size: 16 debug: false @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/BJTaxi-InFlow.yaml b/config/MegaCRN/BJTaxi-InFlow.yaml index c1e5954..b6b0fd5 100644 --- a/config/MegaCRN/BJTaxi-InFlow.yaml +++ b/config/MegaCRN/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: MegaCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 3 cl_decay_steps: 2000 @@ -23,10 +25,12 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 1024 output_dim: 1 rnn_units: 64 use_curriculum_learning: true ycov_dim: 1 + train: batch_size: 32 debug: false @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/BJTaxi-OutFlow.yaml b/config/MegaCRN/BJTaxi-OutFlow.yaml index df43640..41602a5 100644 --- a/config/MegaCRN/BJTaxi-OutFlow.yaml +++ b/config/MegaCRN/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: MegaCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 3 cl_decay_steps: 2000 @@ -23,10 +25,12 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 1024 output_dim: 1 rnn_units: 64 use_curriculum_learning: true ycov_dim: 1 + train: batch_size: 32 debug: false @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/METR-LA.yaml b/config/MegaCRN/METR-LA.yaml index 9be97b9..c3e7805 100644 --- a/config/MegaCRN/METR-LA.yaml +++ b/config/MegaCRN/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: MegaCRN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 3 cl_decay_steps: 2000 @@ -23,10 +25,12 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 207 output_dim: 1 rnn_units: 64 use_curriculum_learning: true ycov_dim: 1 + train: batch_size: 16 debug: false @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/NYCBike-InFlow.yaml b/config/MegaCRN/NYCBike-InFlow.yaml index ef35650..de90784 100644 --- a/config/MegaCRN/NYCBike-InFlow.yaml +++ b/config/MegaCRN/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: MegaCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 3 cl_decay_steps: 2000 @@ -23,10 +25,12 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 128 output_dim: 1 rnn_units: 64 use_curriculum_learning: true ycov_dim: 1 + train: batch_size: 32 debug: false @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/NYCBike-OutFlow.yaml b/config/MegaCRN/NYCBike-OutFlow.yaml index 85465f7..ec0487b 100644 --- a/config/MegaCRN/NYCBike-OutFlow.yaml +++ b/config/MegaCRN/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: MegaCRN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 3 cl_decay_steps: 2000 @@ -23,10 +25,12 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 128 output_dim: 1 rnn_units: 64 use_curriculum_learning: true ycov_dim: 1 + train: batch_size: 32 debug: false @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/PEMSD3.yaml b/config/MegaCRN/PEMSD3.yaml index 2716192..5814af0 100644 --- a/config/MegaCRN/PEMSD3.yaml +++ b/config/MegaCRN/PEMSD3.yaml @@ -25,6 +25,7 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 358 output_dim: 1 rnn_units: 64 use_curriculum_learning: true @@ -43,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/PEMSD4.yaml b/config/MegaCRN/PEMSD4.yaml index 2ed68ca..d3c06c9 100644 --- a/config/MegaCRN/PEMSD4.yaml +++ b/config/MegaCRN/PEMSD4.yaml @@ -25,6 +25,7 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 307 output_dim: 1 rnn_units: 64 use_curriculum_learning: true @@ -43,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/PEMSD7.yaml b/config/MegaCRN/PEMSD7.yaml index 47e34f4..b83d7b3 100644 --- a/config/MegaCRN/PEMSD7.yaml +++ b/config/MegaCRN/PEMSD7.yaml @@ -25,6 +25,7 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 883 output_dim: 1 rnn_units: 64 use_curriculum_learning: true @@ -43,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/PEMSD8.yaml b/config/MegaCRN/PEMSD8.yaml index aeda484..ae40736 100644 --- a/config/MegaCRN/PEMSD8.yaml +++ b/config/MegaCRN/PEMSD8.yaml @@ -25,6 +25,7 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 170 output_dim: 1 rnn_units: 64 use_curriculum_learning: true @@ -43,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/MegaCRN/SolarEnergy.yaml b/config/MegaCRN/SolarEnergy.yaml index ae10bdc..669c0c8 100644 --- a/config/MegaCRN/SolarEnergy.yaml +++ b/config/MegaCRN/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: MegaCRN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 3 cl_decay_steps: 2000 @@ -23,10 +25,12 @@ model: mem_dim: 64 mem_num: 20 num_layers: 1 + num_nodes: 137 output_dim: 137 rnn_units: 64 use_curriculum_learning: true ycov_dim: 1 + train: batch_size: 16 debug: false @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/AirQuality.yaml b/config/NLT/AirQuality.yaml index e5c6a67..c6fa211 100644 --- a/config/NLT/AirQuality.yaml +++ b/config/NLT/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: NLT seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: embed_dim: 10 feature_dim: 6 @@ -26,10 +28,12 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 12 output_dim: 6 output_window: 24 use_day: false use_week: false + train: batch_size: 16 debug: false @@ -43,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/BJTaxi-InFlow.yaml b/config/NLT/BJTaxi-InFlow.yaml index 8f54e18..8c918ab 100644 --- a/config/NLT/BJTaxi-InFlow.yaml +++ b/config/NLT/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: NLT seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: embed_dim: 10 feature_dim: 1 @@ -26,10 +28,12 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 1024 output_dim: 1 output_window: 24 use_day: false use_week: false + train: batch_size: 32 debug: false @@ -43,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/BJTaxi-OutFlow.yaml b/config/NLT/BJTaxi-OutFlow.yaml index 5e989ba..e537d52 100644 --- a/config/NLT/BJTaxi-OutFlow.yaml +++ b/config/NLT/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: NLT seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: embed_dim: 10 feature_dim: 1 @@ -26,10 +28,12 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 1024 output_dim: 1 output_window: 24 use_day: false use_week: false + train: batch_size: 32 debug: false @@ -43,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/METR-LA.yaml b/config/NLT/METR-LA.yaml index bcfc403..03601e9 100644 --- a/config/NLT/METR-LA.yaml +++ b/config/NLT/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: NLT seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: embed_dim: 10 feature_dim: 1 @@ -26,10 +28,12 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 207 output_dim: 1 output_window: 12 use_day: false use_week: false + train: batch_size: 16 debug: false @@ -43,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/NYCBike-InFlow.yaml b/config/NLT/NYCBike-InFlow.yaml index b6ac09a..bde93dc 100644 --- a/config/NLT/NYCBike-InFlow.yaml +++ b/config/NLT/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: NLT seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: embed_dim: 10 feature_dim: 1 @@ -26,10 +28,12 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 128 output_dim: 1 output_window: 24 use_day: false use_week: false + train: batch_size: 32 debug: false @@ -43,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/NYCBike-OutFlow.yaml b/config/NLT/NYCBike-OutFlow.yaml index 5e801b2..8c24df4 100644 --- a/config/NLT/NYCBike-OutFlow.yaml +++ b/config/NLT/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: NLT seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: embed_dim: 10 feature_dim: 1 @@ -26,10 +28,12 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 128 output_dim: 1 output_window: 24 use_day: false use_week: false + train: batch_size: 32 debug: false @@ -43,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/PEMSD3.yaml b/config/NLT/PEMSD3.yaml index 7212f76..8086056 100755 --- a/config/NLT/PEMSD3.yaml +++ b/config/NLT/PEMSD3.yaml @@ -28,6 +28,7 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 358 output_dim: 1 output_window: 12 use_day: false @@ -46,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/PEMSD4.yaml b/config/NLT/PEMSD4.yaml index b924b9f..6d41a7c 100755 --- a/config/NLT/PEMSD4.yaml +++ b/config/NLT/PEMSD4.yaml @@ -28,6 +28,7 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 307 output_dim: 1 output_window: 12 use_day: false @@ -46,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/PEMSD7.yaml b/config/NLT/PEMSD7.yaml index c07708c..7a9783d 100755 --- a/config/NLT/PEMSD7.yaml +++ b/config/NLT/PEMSD7.yaml @@ -28,6 +28,7 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 883 output_dim: 1 output_window: 12 use_day: false @@ -46,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/PEMSD8.yaml b/config/NLT/PEMSD8.yaml index 9a3441f..de8494b 100755 --- a/config/NLT/PEMSD8.yaml +++ b/config/NLT/PEMSD8.yaml @@ -28,6 +28,7 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 170 output_dim: 1 output_window: 12 use_day: false @@ -46,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/NLT/SolarEnergy.yaml b/config/NLT/SolarEnergy.yaml index 9bcd7af..f9da4af 100644 --- a/config/NLT/SolarEnergy.yaml +++ b/config/NLT/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: NLT seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: embed_dim: 10 feature_dim: 137 @@ -26,10 +28,12 @@ model: natt_hops: 4 nfc: 256 num_layers: 2 + num_nodes: 137 output_dim: 137 output_window: 24 use_day: false use_week: false + train: batch_size: 16 debug: false @@ -43,11 +47,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/AirQuality.yaml b/config/PDG2SEQ/AirQuality.yaml index 27ec4a2..a2ad31c 100644 --- a/config/PDG2SEQ/AirQuality.yaml +++ b/config/PDG2SEQ/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: PDG2SEQ seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 12 @@ -23,11 +25,13 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 12 output_dim: 6 rnn_units: 64 time_dim: 8 use_day: true use_week: true + train: batch_size: 16 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/BJTaxi-InFlow.yaml b/config/PDG2SEQ/BJTaxi-InFlow.yaml index 5cbdf37..917b505 100644 --- a/config/PDG2SEQ/BJTaxi-InFlow.yaml +++ b/config/PDG2SEQ/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: PDG2SEQ seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 12 @@ -23,11 +25,13 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 1024 output_dim: 1 rnn_units: 64 time_dim: 8 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/BJTaxi-OutFlow.yaml b/config/PDG2SEQ/BJTaxi-OutFlow.yaml index f50e98a..a5ccc47 100644 --- a/config/PDG2SEQ/BJTaxi-OutFlow.yaml +++ b/config/PDG2SEQ/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: PDG2SEQ seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 12 @@ -23,11 +25,13 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 1024 output_dim: 1 rnn_units: 64 time_dim: 8 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/METR-LA.yaml b/config/PDG2SEQ/METR-LA.yaml index 0a52a6c..6999a03 100644 --- a/config/PDG2SEQ/METR-LA.yaml +++ b/config/PDG2SEQ/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: PDG2SEQ seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 12 @@ -23,11 +25,13 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 207 output_dim: 1 rnn_units: 64 time_dim: 8 use_day: true use_week: true + train: batch_size: 16 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/NYCBike-InFlow.yaml b/config/PDG2SEQ/NYCBike-InFlow.yaml index d898dcc..56c3abe 100644 --- a/config/PDG2SEQ/NYCBike-InFlow.yaml +++ b/config/PDG2SEQ/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: PDG2SEQ seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 12 @@ -23,11 +25,13 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 128 output_dim: 1 rnn_units: 64 time_dim: 8 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/NYCBike-OutFlow.yaml b/config/PDG2SEQ/NYCBike-OutFlow.yaml index 52dee49..39dc207 100644 --- a/config/PDG2SEQ/NYCBike-OutFlow.yaml +++ b/config/PDG2SEQ/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: PDG2SEQ seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 12 @@ -23,11 +25,13 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 128 output_dim: 1 rnn_units: 64 time_dim: 8 use_day: true use_week: true + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/PEMSD3.yaml b/config/PDG2SEQ/PEMSD3.yaml index 015116b..f0e6730 100755 --- a/config/PDG2SEQ/PEMSD3.yaml +++ b/config/PDG2SEQ/PEMSD3.yaml @@ -25,6 +25,7 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 358 output_dim: 1 rnn_units: 64 time_dim: 8 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/PEMSD4.yaml b/config/PDG2SEQ/PEMSD4.yaml index a4cb033..3f28b2e 100755 --- a/config/PDG2SEQ/PEMSD4.yaml +++ b/config/PDG2SEQ/PEMSD4.yaml @@ -25,6 +25,7 @@ model: lr_decay_step: 1500 lr_decay_step1: 60,75,90,120 num_layers: 1 + num_nodes: 307 output_dim: 1 rnn_units: 64 time_dim: 16 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/PEMSD7.yaml b/config/PDG2SEQ/PEMSD7.yaml index 5cd0707..4922dfc 100755 --- a/config/PDG2SEQ/PEMSD7.yaml +++ b/config/PDG2SEQ/PEMSD7.yaml @@ -25,6 +25,7 @@ model: lr_decay_step: 12000 lr_decay_step1: 80,100,120 num_layers: 1 + num_nodes: 883 output_dim: 1 rnn_units: 64 time_dim: 20 @@ -54,5 +55,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/PDG2SEQ/PEMSD8.yaml b/config/PDG2SEQ/PEMSD8.yaml index f250216..3c25095 100755 --- a/config/PDG2SEQ/PEMSD8.yaml +++ b/config/PDG2SEQ/PEMSD8.yaml @@ -25,6 +25,7 @@ model: lr_decay_step: 2000 lr_decay_step1: 50,75 num_layers: 1 + num_nodes: 170 output_dim: 1 rnn_units: 64 time_dim: 16 @@ -50,5 +51,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 12 weight_decay: 0 diff --git a/config/PDG2SEQ/SolarEnergy.yaml b/config/PDG2SEQ/SolarEnergy.yaml index 42c6d14..a04a56f 100644 --- a/config/PDG2SEQ/SolarEnergy.yaml +++ b/config/PDG2SEQ/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: PDG2SEQ seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 12 @@ -23,11 +25,13 @@ model: lr_decay_step: 10000 lr_decay_step1: 75,90,120 num_layers: 1 + num_nodes: 137 output_dim: 137 rnn_units: 64 time_dim: 8 use_day: true use_week: true + train: batch_size: 16 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/REPST/AirQuality.yaml b/config/REPST/AirQuality.yaml index f192382..a40e11e 100755 --- a/config/REPST/AirQuality.yaml +++ b/config/REPST/AirQuality.yaml @@ -13,7 +13,7 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 @@ -26,7 +26,7 @@ model: gpt_path: ./GPT-2 input_dim: 6 n_heads: 1 - num_nodes: 35 + num_nodes: 12 output_dim: 3 patch_len: 6 pred_len: 24 diff --git a/config/REPST/BJTaxi-InFlow.yaml b/config/REPST/BJTaxi-InFlow.yaml index 56ccf66..3191eba 100644 --- a/config/REPST/BJTaxi-InFlow.yaml +++ b/config/REPST/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: REPST seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -31,6 +33,7 @@ model: seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 16 debug: false diff --git a/config/REPST/BJTaxi-OutFlow.yaml b/config/REPST/BJTaxi-OutFlow.yaml index 36dae39..2c251e6 100644 --- a/config/REPST/BJTaxi-OutFlow.yaml +++ b/config/REPST/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: REPST seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -31,6 +33,7 @@ model: seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 16 debug: false diff --git a/config/REPST/NYCBike-InFlow.yaml b/config/REPST/NYCBike-InFlow.yaml index b63b151..3ed89c8 100644 --- a/config/REPST/NYCBike-InFlow.yaml +++ b/config/REPST/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: REPST seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -24,13 +26,14 @@ model: gpt_path: ./GPT-2 input_dim: 1 n_heads: 1 - num_nodes: 1024 + num_nodes: 128 output_dim: 1 patch_len: 6 pred_len: 24 seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 16 debug: false diff --git a/config/REPST/NYCBike-OutFlow.yaml b/config/REPST/NYCBike-OutFlow.yaml index 9ab3c6d..59d4364 100644 --- a/config/REPST/NYCBike-OutFlow.yaml +++ b/config/REPST/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: REPST seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_ff: 128 d_model: 64 @@ -24,13 +26,14 @@ model: gpt_path: ./GPT-2 input_dim: 1 n_heads: 1 - num_nodes: 1024 + num_nodes: 128 output_dim: 1 patch_len: 6 pred_len: 24 seq_len: 24 stride: 7 word_num: 1000 + train: batch_size: 16 debug: false diff --git a/config/STAEFormer/AirQuality.yaml b/config/STAEFormer/AirQuality.yaml index b622adc..e5f07e8 100644 --- a/config/STAEFormer/AirQuality.yaml +++ b/config/STAEFormer/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAEFormer seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: adaptive_embedding_dim: 80 dow_embedding_dim: 24 @@ -26,13 +28,14 @@ model: input_embedding_dim: 24 num_heads: 4 num_layers: 3 - num_nodes: 35 + num_nodes: 12 out_steps: 24 output_dim: 6 spatial_embedding_dim: 0 steps_per_day: 24 tod_embedding_dim: 24 use_mixed_proj: true + train: batch_size: 16 debug: false @@ -46,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/BJTaxi-InFlow.yaml b/config/STAEFormer/BJTaxi-InFlow.yaml index 3404e8a..7eb24c1 100644 --- a/config/STAEFormer/BJTaxi-InFlow.yaml +++ b/config/STAEFormer/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAEFormer seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adaptive_embedding_dim: 80 dow_embedding_dim: 24 @@ -33,6 +35,7 @@ model: steps_per_day: 48 tod_embedding_dim: 24 use_mixed_proj: true + train: batch_size: 32 debug: false @@ -46,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/BJTaxi-OutFlow.yaml b/config/STAEFormer/BJTaxi-OutFlow.yaml index 76c7369..fbc5d56 100644 --- a/config/STAEFormer/BJTaxi-OutFlow.yaml +++ b/config/STAEFormer/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAEFormer seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adaptive_embedding_dim: 80 dow_embedding_dim: 24 @@ -33,6 +35,7 @@ model: steps_per_day: 48 tod_embedding_dim: 24 use_mixed_proj: true + train: batch_size: 32 debug: false @@ -46,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/METR-LA.yaml b/config/STAEFormer/METR-LA.yaml index e982b2b..003e50e 100644 --- a/config/STAEFormer/METR-LA.yaml +++ b/config/STAEFormer/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAEFormer seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: adaptive_embedding_dim: 80 dow_embedding_dim: 24 @@ -33,6 +35,7 @@ model: steps_per_day: 288 tod_embedding_dim: 24 use_mixed_proj: true + train: batch_size: 16 debug: false @@ -46,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/NYCBike-InFlow.yaml b/config/STAEFormer/NYCBike-InFlow.yaml index 4f88780..a96571c 100644 --- a/config/STAEFormer/NYCBike-InFlow.yaml +++ b/config/STAEFormer/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAEFormer seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adaptive_embedding_dim: 80 dow_embedding_dim: 24 @@ -26,13 +28,14 @@ model: input_embedding_dim: 24 num_heads: 4 num_layers: 3 - num_nodes: 1024 + num_nodes: 128 out_steps: 24 output_dim: 1 spatial_embedding_dim: 0 steps_per_day: 48 tod_embedding_dim: 24 use_mixed_proj: true + train: batch_size: 32 debug: false @@ -46,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/NYCBike-OutFlow.yaml b/config/STAEFormer/NYCBike-OutFlow.yaml index ee13784..dc9d430 100644 --- a/config/STAEFormer/NYCBike-OutFlow.yaml +++ b/config/STAEFormer/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAEFormer seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adaptive_embedding_dim: 80 dow_embedding_dim: 24 @@ -26,13 +28,14 @@ model: input_embedding_dim: 24 num_heads: 4 num_layers: 3 - num_nodes: 1024 + num_nodes: 128 out_steps: 24 output_dim: 1 spatial_embedding_dim: 0 steps_per_day: 48 tod_embedding_dim: 24 use_mixed_proj: true + train: batch_size: 32 debug: false @@ -46,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/PEMSD3.yaml b/config/STAEFormer/PEMSD3.yaml index 79eb4de..7497a8b 100755 --- a/config/STAEFormer/PEMSD3.yaml +++ b/config/STAEFormer/PEMSD3.yaml @@ -49,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/PEMSD4.yaml b/config/STAEFormer/PEMSD4.yaml index a832b53..b248ffc 100755 --- a/config/STAEFormer/PEMSD4.yaml +++ b/config/STAEFormer/PEMSD4.yaml @@ -49,10 +49,9 @@ train: lr_decay_rate: 0.1 lr_decay_step: 5,20,40,70 lr_init: 0.001 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0.0003 diff --git a/config/STAEFormer/PEMSD7.yaml b/config/STAEFormer/PEMSD7.yaml index e41e643..be99282 100755 --- a/config/STAEFormer/PEMSD7.yaml +++ b/config/STAEFormer/PEMSD7.yaml @@ -49,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/PEMSD8.yaml b/config/STAEFormer/PEMSD8.yaml index dbee2c7..d9c91a9 100755 --- a/config/STAEFormer/PEMSD8.yaml +++ b/config/STAEFormer/PEMSD8.yaml @@ -49,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAEFormer/SolarEnergy.yaml b/config/STAEFormer/SolarEnergy.yaml index fafffd6..fd97a63 100644 --- a/config/STAEFormer/SolarEnergy.yaml +++ b/config/STAEFormer/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAEFormer seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: adaptive_embedding_dim: 80 dow_embedding_dim: 24 @@ -33,6 +35,7 @@ model: steps_per_day: 24 tod_embedding_dim: 24 use_mixed_proj: true + train: batch_size: 16 debug: false @@ -46,11 +49,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/AirQuality.yaml b/config/STAWnet/AirQuality.yaml index 8f9f94f..6d3e0d0 100644 --- a/config/STAWnet/AirQuality.yaml +++ b/config/STAWnet/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAWnet seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: addaptadj: true aptonly: false @@ -30,9 +32,11 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 12 output_dim: 6 residual_channels: 32 skip_channels: 256 + train: batch_size: 16 debug: false @@ -46,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/BJTaxi-InFlow.yaml b/config/STAWnet/BJTaxi-InFlow.yaml index 029930a..edd919a 100644 --- a/config/STAWnet/BJTaxi-InFlow.yaml +++ b/config/STAWnet/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAWnet seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: addaptadj: true aptonly: false @@ -30,9 +32,11 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 1024 output_dim: 1 residual_channels: 32 skip_channels: 256 + train: batch_size: 32 debug: false @@ -46,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/BJTaxi-OutFlow.yaml b/config/STAWnet/BJTaxi-OutFlow.yaml index f3856e8..e40975b 100644 --- a/config/STAWnet/BJTaxi-OutFlow.yaml +++ b/config/STAWnet/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAWnet seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: addaptadj: true aptonly: false @@ -30,9 +32,11 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 1024 output_dim: 1 residual_channels: 32 skip_channels: 256 + train: batch_size: 32 debug: false @@ -46,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/METR-LA.yaml b/config/STAWnet/METR-LA.yaml index dc84df8..d0fc158 100644 --- a/config/STAWnet/METR-LA.yaml +++ b/config/STAWnet/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAWnet seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: addaptadj: true aptonly: false @@ -30,9 +32,11 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 207 output_dim: 1 residual_channels: 32 skip_channels: 256 + train: batch_size: 16 debug: false @@ -46,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/NYCBike-InFlow.yaml b/config/STAWnet/NYCBike-InFlow.yaml index caea941..563d80c 100644 --- a/config/STAWnet/NYCBike-InFlow.yaml +++ b/config/STAWnet/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAWnet seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: addaptadj: true aptonly: false @@ -30,9 +32,11 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 128 output_dim: 1 residual_channels: 32 skip_channels: 256 + train: batch_size: 32 debug: false @@ -46,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/NYCBike-OutFlow.yaml b/config/STAWnet/NYCBike-OutFlow.yaml index 33a377e..38853aa 100644 --- a/config/STAWnet/NYCBike-OutFlow.yaml +++ b/config/STAWnet/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAWnet seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: addaptadj: true aptonly: false @@ -30,9 +32,11 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 128 output_dim: 1 residual_channels: 32 skip_channels: 256 + train: batch_size: 32 debug: false @@ -46,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/PEMSD3.yaml b/config/STAWnet/PEMSD3.yaml index 30aaddc..9b0f48f 100644 --- a/config/STAWnet/PEMSD3.yaml +++ b/config/STAWnet/PEMSD3.yaml @@ -32,6 +32,7 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 358 output_dim: 1 residual_channels: 32 skip_channels: 256 @@ -49,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/PEMSD4.yaml b/config/STAWnet/PEMSD4.yaml index b89454f..d17a5ab 100644 --- a/config/STAWnet/PEMSD4.yaml +++ b/config/STAWnet/PEMSD4.yaml @@ -32,6 +32,7 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 307 output_dim: 1 residual_channels: 32 skip_channels: 256 @@ -49,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/PEMSD7.yaml b/config/STAWnet/PEMSD7.yaml index 5c52d57..c018131 100644 --- a/config/STAWnet/PEMSD7.yaml +++ b/config/STAWnet/PEMSD7.yaml @@ -32,6 +32,7 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 883 output_dim: 1 residual_channels: 32 skip_channels: 256 @@ -49,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/PEMSD8.yaml b/config/STAWnet/PEMSD8.yaml index 52fcddf..a0f9e0a 100644 --- a/config/STAWnet/PEMSD8.yaml +++ b/config/STAWnet/PEMSD8.yaml @@ -32,6 +32,7 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 170 output_dim: 1 residual_channels: 32 skip_channels: 256 @@ -49,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STAWnet/SolarEnergy.yaml b/config/STAWnet/SolarEnergy.yaml index d4e3b0a..6a2cfb3 100644 --- a/config/STAWnet/SolarEnergy.yaml +++ b/config/STAWnet/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STAWnet seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: addaptadj: true aptonly: false @@ -30,9 +32,11 @@ model: kernel_size: 2 layers: 2 noapt: false + num_nodes: 137 output_dim: 137 residual_channels: 32 skip_channels: 256 + train: batch_size: 16 debug: false @@ -46,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/AirQuality.yaml b/config/STFGNN/AirQuality.yaml index 34ab48b..d559c4e 100644 --- a/config/STFGNN/AirQuality.yaml +++ b/config/STFGNN/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STFGNN seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,31 +13,34 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 24 input_dim: 6 mask: None + num_nodes: 12 out_layer_dim: 128 output_dim: 6 spatial_emb: true temporal_emb: true window: 24 + train: batch_size: 16 debug: false @@ -50,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/BJTaxi-InFlow.yaml b/config/STFGNN/BJTaxi-InFlow.yaml index ca1d078..0b9f284 100644 --- a/config/STFGNN/BJTaxi-InFlow.yaml +++ b/config/STFGNN/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STFGNN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,27 +17,30 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 24 input_dim: 1 mask: None + num_nodes: 1024 out_layer_dim: 128 output_dim: 1 spatial_emb: true temporal_emb: true window: 24 + train: batch_size: 32 debug: false @@ -50,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/BJTaxi-OutFlow.yaml b/config/STFGNN/BJTaxi-OutFlow.yaml index 32e5a5c..75916dd 100644 --- a/config/STFGNN/BJTaxi-OutFlow.yaml +++ b/config/STFGNN/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STFGNN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,27 +17,30 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 24 input_dim: 1 mask: None + num_nodes: 1024 out_layer_dim: 128 output_dim: 1 spatial_emb: true temporal_emb: true window: 24 + train: batch_size: 32 debug: false @@ -50,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/METR-LA.yaml b/config/STFGNN/METR-LA.yaml index 2f39be2..5553fd0 100644 --- a/config/STFGNN/METR-LA.yaml +++ b/config/STFGNN/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STFGNN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,27 +17,30 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 12 input_dim: 1 mask: None + num_nodes: 207 out_layer_dim: 128 output_dim: 1 spatial_emb: true temporal_emb: true window: 12 + train: batch_size: 16 debug: false @@ -50,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/NYCBike-InFlow.yaml b/config/STFGNN/NYCBike-InFlow.yaml index 7c123a3..0a903f9 100644 --- a/config/STFGNN/NYCBike-InFlow.yaml +++ b/config/STFGNN/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STFGNN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,31 +13,34 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 24 input_dim: 1 mask: None + num_nodes: 128 out_layer_dim: 128 output_dim: 1 spatial_emb: true temporal_emb: true window: 24 + train: batch_size: 32 debug: false @@ -50,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/NYCBike-OutFlow.yaml b/config/STFGNN/NYCBike-OutFlow.yaml index d170b59..199c36f 100644 --- a/config/STFGNN/NYCBike-OutFlow.yaml +++ b/config/STFGNN/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STFGNN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,31 +13,34 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 24 input_dim: 1 mask: None + num_nodes: 128 out_layer_dim: 128 output_dim: 1 spatial_emb: true temporal_emb: true window: 24 + train: batch_size: 32 debug: false @@ -50,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/PEMSD3.yaml b/config/STFGNN/PEMSD3.yaml index 35312b4..0935f72 100755 --- a/config/STFGNN/PEMSD3.yaml +++ b/config/STFGNN/PEMSD3.yaml @@ -21,10 +21,20 @@ data: model: activation: GLU first_layer_embedding_size: 64 - hidden_dims: [[64, 64, 64], [64, 64, 64], [64, 64, 64]] + hidden_dims: + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 12 input_dim: 1 mask: None + num_nodes: 358 out_layer_dim: 128 output_dim: 1 spatial_emb: true @@ -44,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/PEMSD4.yaml b/config/STFGNN/PEMSD4.yaml index 8df35b7..b47b851 100755 --- a/config/STFGNN/PEMSD4.yaml +++ b/config/STFGNN/PEMSD4.yaml @@ -21,10 +21,20 @@ data: model: activation: GLU first_layer_embedding_size: 64 - hidden_dims: [[64, 64, 64], [64, 64, 64], [64, 64, 64]] + hidden_dims: + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 12 input_dim: 1 mask: None + num_nodes: 307 out_layer_dim: 128 output_dim: 1 spatial_emb: true @@ -44,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/PEMSD7.yaml b/config/STFGNN/PEMSD7.yaml index d5338f8..b7320f6 100755 --- a/config/STFGNN/PEMSD7.yaml +++ b/config/STFGNN/PEMSD7.yaml @@ -21,10 +21,20 @@ data: model: activation: GLU first_layer_embedding_size: 64 - hidden_dims: [[32, 32, 32], [32, 32, 32], [32, 32, 32]] + hidden_dims: + - - 32 + - 32 + - 32 + - - 32 + - 32 + - 32 + - - 32 + - 32 + - 32 horizon: 12 input_dim: 1 mask: None + num_nodes: 883 out_layer_dim: 64 output_dim: 1 spatial_emb: true @@ -45,11 +55,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/PEMSD8.yaml b/config/STFGNN/PEMSD8.yaml index a1f49d4..08d4f72 100755 --- a/config/STFGNN/PEMSD8.yaml +++ b/config/STFGNN/PEMSD8.yaml @@ -21,10 +21,20 @@ data: model: activation: GLU first_layer_embedding_size: 64 - hidden_dims: [[64, 64, 64], [64, 64, 64], [64, 64, 64]] + hidden_dims: + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 12 input_dim: 1 mask: None + num_nodes: 170 out_layer_dim: 128 output_dim: 1 spatial_emb: true @@ -44,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STFGNN/SolarEnergy.yaml b/config/STFGNN/SolarEnergy.yaml index 5f5f052..2531a1a 100644 --- a/config/STFGNN/SolarEnergy.yaml +++ b/config/STFGNN/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STFGNN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,27 +17,30 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 horizon: 24 input_dim: 137 mask: None + num_nodes: 137 out_layer_dim: 128 output_dim: 137 spatial_emb: true temporal_emb: true window: 24 + train: batch_size: 16 debug: false @@ -50,11 +54,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGCN/AirQuality.yaml b/config/STGCN/AirQuality.yaml index 4a5b272..4c684d3 100644 --- a/config/STGCN/AirQuality.yaml +++ b/config/STGCN/AirQuality.yaml @@ -13,7 +13,7 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 6 n_his: 24 + num_nodes: 12 output_dim: 6 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/STGCN/BJTaxi-InFlow.yaml b/config/STGCN/BJTaxi-InFlow.yaml index 8a8cc89..6860b15 100644 --- a/config/STGCN/BJTaxi-InFlow.yaml +++ b/config/STGCN/BJTaxi-InFlow.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 24 + num_nodes: 1024 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/STGCN/BJTaxi-OutFlow.yaml b/config/STGCN/BJTaxi-OutFlow.yaml index b0641dc..8480e65 100644 --- a/config/STGCN/BJTaxi-OutFlow.yaml +++ b/config/STGCN/BJTaxi-OutFlow.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 24 + num_nodes: 1024 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/STGCN/METR-LA.yaml b/config/STGCN/METR-LA.yaml index e73ecbf..fe24edc 100644 --- a/config/STGCN/METR-LA.yaml +++ b/config/STGCN/METR-LA.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 24 + num_nodes: 207 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/STGCN/NYCBike-InFlow.yaml b/config/STGCN/NYCBike-InFlow.yaml index b01cd36..29a07a6 100644 --- a/config/STGCN/NYCBike-InFlow.yaml +++ b/config/STGCN/NYCBike-InFlow.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 24 + num_nodes: 128 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/STGCN/NYCBike-OutFlow.yaml b/config/STGCN/NYCBike-OutFlow.yaml index 8171033..1c747eb 100644 --- a/config/STGCN/NYCBike-OutFlow.yaml +++ b/config/STGCN/NYCBike-OutFlow.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 24 + num_nodes: 128 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/STGCN/PEMSD3.yaml b/config/STGCN/PEMSD3.yaml index ab254ec..6fa6c75 100755 --- a/config/STGCN/PEMSD3.yaml +++ b/config/STGCN/PEMSD3.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 12 + num_nodes: 358 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGCN/PEMSD4.yaml b/config/STGCN/PEMSD4.yaml index bc62528..596f6f2 100755 --- a/config/STGCN/PEMSD4.yaml +++ b/config/STGCN/PEMSD4.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 12 + num_nodes: 307 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGCN/PEMSD7.yaml b/config/STGCN/PEMSD7.yaml index 1f4139b..9615f85 100755 --- a/config/STGCN/PEMSD7.yaml +++ b/config/STGCN/PEMSD7.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 12 + num_nodes: 883 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGCN/PEMSD8.yaml b/config/STGCN/PEMSD8.yaml index 2dd7bce..846ad3f 100755 --- a/config/STGCN/PEMSD8.yaml +++ b/config/STGCN/PEMSD8.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 12 + num_nodes: 170 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGCN/SolarEnergy.yaml b/config/STGCN/SolarEnergy.yaml index fc9ecc7..d3e44dc 100644 --- a/config/STGCN/SolarEnergy.yaml +++ b/config/STGCN/SolarEnergy.yaml @@ -28,6 +28,7 @@ model: gso_type: sym_norm_lap input_dim: 1 n_his: 24 + num_nodes: 137 output_dim: 1 stblock_num: 2 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/STGNCDE/AirQuality.yaml b/config/STGNCDE/AirQuality.yaml index c7905f6..88bf8f0 100644 --- a/config/STGNCDE/AirQuality.yaml +++ b/config/STGNCDE/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNCDE seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 10 @@ -24,9 +26,11 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 12 output_dim: 6 solver: rk4 type: type1 + train: batch_size: 16 debug: false @@ -39,16 +43,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/BJTaxi-InFlow.yaml b/config/STGNCDE/BJTaxi-InFlow.yaml index 0bb3ab5..0de1907 100644 --- a/config/STGNCDE/BJTaxi-InFlow.yaml +++ b/config/STGNCDE/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNCDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 10 @@ -24,9 +26,11 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 1024 output_dim: 1 solver: rk4 type: type1 + train: batch_size: 32 debug: false @@ -39,16 +43,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/BJTaxi-OutFlow.yaml b/config/STGNCDE/BJTaxi-OutFlow.yaml index 4cc5fdb..2022544 100644 --- a/config/STGNCDE/BJTaxi-OutFlow.yaml +++ b/config/STGNCDE/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNCDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 10 @@ -24,9 +26,11 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 1024 output_dim: 1 solver: rk4 type: type1 + train: batch_size: 32 debug: false @@ -39,16 +43,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/METR-LA.yaml b/config/STGNCDE/METR-LA.yaml index 135de6f..127e6fa 100644 --- a/config/STGNCDE/METR-LA.yaml +++ b/config/STGNCDE/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNCDE seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 10 @@ -24,9 +26,11 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 207 output_dim: 1 solver: rk4 type: type1 + train: batch_size: 16 debug: false @@ -39,16 +43,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/NYCBike-InFlow.yaml b/config/STGNCDE/NYCBike-InFlow.yaml index a35aeb5..0d00183 100644 --- a/config/STGNCDE/NYCBike-InFlow.yaml +++ b/config/STGNCDE/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNCDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 10 @@ -24,9 +26,11 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 128 output_dim: 1 solver: rk4 type: type1 + train: batch_size: 32 debug: false @@ -39,16 +43,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/NYCBike-OutFlow.yaml b/config/STGNCDE/NYCBike-OutFlow.yaml index 98d94a2..b54a641 100644 --- a/config/STGNCDE/NYCBike-OutFlow.yaml +++ b/config/STGNCDE/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNCDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 10 @@ -24,9 +26,11 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 128 output_dim: 1 solver: rk4 type: type1 + train: batch_size: 32 debug: false @@ -39,16 +43,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/PEMSD3.yaml b/config/STGNCDE/PEMSD3.yaml index b6abea6..83e880c 100755 --- a/config/STGNCDE/PEMSD3.yaml +++ b/config/STGNCDE/PEMSD3.yaml @@ -26,6 +26,7 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 358 output_dim: 1 solver: rk4 type: type1 @@ -41,13 +42,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/PEMSD4.yaml b/config/STGNCDE/PEMSD4.yaml index 7c2d8d4..8df1d04 100755 --- a/config/STGNCDE/PEMSD4.yaml +++ b/config/STGNCDE/PEMSD4.yaml @@ -26,6 +26,7 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 307 output_dim: 1 solver: rk4 type: type1 @@ -41,13 +42,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/PEMSD7.yaml b/config/STGNCDE/PEMSD7.yaml index b11d474..ae7bbf5 100755 --- a/config/STGNCDE/PEMSD7.yaml +++ b/config/STGNCDE/PEMSD7.yaml @@ -26,6 +26,7 @@ model: hid_hid_dim: 64 input_dim: 2 num_layers: 2 + num_nodes: 883 output_dim: 1 solver: rk4 type: type1 @@ -41,13 +42,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/PEMSD8.yaml b/config/STGNCDE/PEMSD8.yaml index 452f280..22b22ab 100755 --- a/config/STGNCDE/PEMSD8.yaml +++ b/config/STGNCDE/PEMSD8.yaml @@ -26,6 +26,7 @@ model: hid_hid_dim: 64 input_dim: 2 num_layers: 2 + num_nodes: 170 output_dim: 1 solver: rk4 type: type1 @@ -41,13 +42,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNCDE/SolarEnergy.yaml b/config/STGNCDE/SolarEnergy.yaml index db78adc..134268a 100644 --- a/config/STGNCDE/SolarEnergy.yaml +++ b/config/STGNCDE/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNCDE seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_k: 2 embed_dim: 10 @@ -24,9 +26,11 @@ model: hid_hid_dim: 128 input_dim: 2 num_layers: 2 + num_nodes: 137 output_dim: 137 solver: rk4 type: type1 + train: batch_size: 16 debug: false @@ -39,16 +43,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/AirQuality.yaml b/config/STGNRDE/AirQuality.yaml index 340696a..7086bb6 100644 --- a/config/STGNRDE/AirQuality.yaml +++ b/config/STGNRDE/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNRDE seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: adp_opt: false cheb_k: 3 @@ -28,8 +30,10 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 12 output_dim: 6 solver: rk4 + train: batch_size: 16 debug: false @@ -42,16 +46,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/BJTaxi-InFlow.yaml b/config/STGNRDE/BJTaxi-InFlow.yaml index 891d32b..d1d6275 100644 --- a/config/STGNRDE/BJTaxi-InFlow.yaml +++ b/config/STGNRDE/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNRDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adp_opt: false cheb_k: 3 @@ -28,8 +30,10 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 1024 output_dim: 1 solver: rk4 + train: batch_size: 32 debug: false @@ -42,16 +46,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/BJTaxi-OutFlow.yaml b/config/STGNRDE/BJTaxi-OutFlow.yaml index 8646195..36ec5b2 100644 --- a/config/STGNRDE/BJTaxi-OutFlow.yaml +++ b/config/STGNRDE/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNRDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adp_opt: false cheb_k: 3 @@ -28,8 +30,10 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 1024 output_dim: 1 solver: rk4 + train: batch_size: 32 debug: false @@ -42,16 +46,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/METR-LA.yaml b/config/STGNRDE/METR-LA.yaml index 00b2934..7eafd79 100644 --- a/config/STGNRDE/METR-LA.yaml +++ b/config/STGNRDE/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNRDE seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: adp_opt: false cheb_k: 3 @@ -28,8 +30,10 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 207 output_dim: 1 solver: rk4 + train: batch_size: 16 debug: false @@ -42,16 +46,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/NYCBike-InFlow.yaml b/config/STGNRDE/NYCBike-InFlow.yaml index f25a31b..c204cdb 100644 --- a/config/STGNRDE/NYCBike-InFlow.yaml +++ b/config/STGNRDE/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNRDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adp_opt: false cheb_k: 3 @@ -28,8 +30,10 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 128 output_dim: 1 solver: rk4 + train: batch_size: 32 debug: false @@ -42,16 +46,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/NYCBike-OutFlow.yaml b/config/STGNRDE/NYCBike-OutFlow.yaml index 8a3336d..27d11ee 100644 --- a/config/STGNRDE/NYCBike-OutFlow.yaml +++ b/config/STGNRDE/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNRDE seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: adp_opt: false cheb_k: 3 @@ -28,8 +30,10 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 128 output_dim: 1 solver: rk4 + train: batch_size: 32 debug: false @@ -42,16 +46,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/PEMSD3.yaml b/config/STGNRDE/PEMSD3.yaml index 14a4ba9..af1529a 100644 --- a/config/STGNRDE/PEMSD3.yaml +++ b/config/STGNRDE/PEMSD3.yaml @@ -30,6 +30,7 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 358 output_dim: 1 solver: rk4 @@ -44,13 +45,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/PEMSD4.yaml b/config/STGNRDE/PEMSD4.yaml index aadfe01..b0d392a 100644 --- a/config/STGNRDE/PEMSD4.yaml +++ b/config/STGNRDE/PEMSD4.yaml @@ -30,6 +30,7 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 307 output_dim: 1 solver: rk4 @@ -44,13 +45,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/PEMSD7.yaml b/config/STGNRDE/PEMSD7.yaml index 1d068f0..4b8e399 100644 --- a/config/STGNRDE/PEMSD7.yaml +++ b/config/STGNRDE/PEMSD7.yaml @@ -30,6 +30,7 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 883 output_dim: 1 solver: rk4 @@ -44,13 +45,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/PEMSD8.yaml b/config/STGNRDE/PEMSD8.yaml index 66c53bd..e765d25 100644 --- a/config/STGNRDE/PEMSD8.yaml +++ b/config/STGNRDE/PEMSD8.yaml @@ -30,6 +30,7 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 170 output_dim: 1 solver: rk4 @@ -44,13 +45,16 @@ 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: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGNRDE/SolarEnergy.yaml b/config/STGNRDE/SolarEnergy.yaml index c6a96b7..a9512d8 100644 --- a/config/STGNRDE/SolarEnergy.yaml +++ b/config/STGNRDE/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGNRDE seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: adp_opt: false cheb_k: 3 @@ -28,8 +30,10 @@ model: interpolation: cubic model_type: rde num_layers: 2 + num_nodes: 137 output_dim: 137 solver: rk4 + train: batch_size: 16 debug: false @@ -42,16 +46,15 @@ train: lr_decay: false lr_decay_rate: 0.3 lr_decay_step: - - 5 - - 20 - - 40 - - 70 + - 5 + - 20 + - 40 + - 70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/AirQuality.yaml b/config/STGODE/AirQuality.yaml index 58bd244..14ad5b1 100644 --- a/config/STGODE/AirQuality.yaml +++ b/config/STGODE/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGODE seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,20 +13,23 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: history: 24 horizon: 24 input_dim: 6 num_features: 6 + num_nodes: 12 output_dim: 6 sigma1: 0.1 sigma2: 10 thres1: 0.6 thres2: 0.5 + train: batch_size: 16 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/BJTaxi-InFlow.yaml b/config/STGODE/BJTaxi-InFlow.yaml index d596a5c..5637bf5 100644 --- a/config/STGODE/BJTaxi-InFlow.yaml +++ b/config/STGODE/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGODE seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: history: 24 horizon: 24 input_dim: 1 num_features: 1 + num_nodes: 1024 output_dim: 1 sigma1: 0.1 sigma2: 10 thres1: 0.6 thres2: 0.5 + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/BJTaxi-OutFlow.yaml b/config/STGODE/BJTaxi-OutFlow.yaml index f2a476f..4ee73d3 100644 --- a/config/STGODE/BJTaxi-OutFlow.yaml +++ b/config/STGODE/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGODE seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: history: 24 horizon: 24 input_dim: 1 num_features: 1 + num_nodes: 1024 output_dim: 1 sigma1: 0.1 sigma2: 10 thres1: 0.6 thres2: 0.5 + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/METR-LA.yaml b/config/STGODE/METR-LA.yaml index 4527f0c..895050f 100644 --- a/config/STGODE/METR-LA.yaml +++ b/config/STGODE/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGODE seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: history: 12 horizon: 12 input_dim: 1 num_features: 1 + num_nodes: 207 output_dim: 1 sigma1: 0.1 sigma2: 10 thres1: 0.6 thres2: 0.5 + train: batch_size: 16 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/NYCBike-InFlow.yaml b/config/STGODE/NYCBike-InFlow.yaml index 68f9e95..c8b1757 100644 --- a/config/STGODE/NYCBike-InFlow.yaml +++ b/config/STGODE/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGODE seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,20 +13,23 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: history: 24 horizon: 24 input_dim: 1 num_features: 1 + num_nodes: 128 output_dim: 1 sigma1: 0.1 sigma2: 10 thres1: 0.6 thres2: 0.5 + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/NYCBike-OutFlow.yaml b/config/STGODE/NYCBike-OutFlow.yaml index a4fabdd..858c455 100644 --- a/config/STGODE/NYCBike-OutFlow.yaml +++ b/config/STGODE/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGODE seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,20 +13,23 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: history: 24 horizon: 24 input_dim: 1 num_features: 1 + num_nodes: 128 output_dim: 1 sigma1: 0.1 sigma2: 10 thres1: 0.6 thres2: 0.5 + train: batch_size: 32 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/PEMSD3.yaml b/config/STGODE/PEMSD3.yaml index 11cd3b3..c4b5790 100755 --- a/config/STGODE/PEMSD3.yaml +++ b/config/STGODE/PEMSD3.yaml @@ -23,6 +23,7 @@ model: horizon: 12 input_dim: 1 num_features: 1 + num_nodes: 358 output_dim: 1 sigma1: 0.1 sigma2: 10 @@ -42,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/PEMSD4.yaml b/config/STGODE/PEMSD4.yaml index db52560..ba17e0c 100755 --- a/config/STGODE/PEMSD4.yaml +++ b/config/STGODE/PEMSD4.yaml @@ -23,6 +23,7 @@ model: horizon: 12 input_dim: 1 num_features: 1 + num_nodes: 307 output_dim: 1 sigma1: 0.1 sigma2: 10 @@ -42,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/PEMSD7.yaml b/config/STGODE/PEMSD7.yaml index eadd560..d5f6442 100755 --- a/config/STGODE/PEMSD7.yaml +++ b/config/STGODE/PEMSD7.yaml @@ -23,6 +23,7 @@ model: horizon: 12 input_dim: 1 num_features: 1 + num_nodes: 883 output_dim: 1 sigma1: 0.1 sigma2: 10 @@ -42,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/PEMSD8.yaml b/config/STGODE/PEMSD8.yaml index 70fc37c..19d7e66 100755 --- a/config/STGODE/PEMSD8.yaml +++ b/config/STGODE/PEMSD8.yaml @@ -23,6 +23,7 @@ model: horizon: 12 input_dim: 1 num_features: 1 + num_nodes: 170 output_dim: 1 sigma1: 0.1 sigma2: 10 @@ -42,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STGODE/SolarEnergy.yaml b/config/STGODE/SolarEnergy.yaml index 304df8d..1275bdc 100644 --- a/config/STGODE/SolarEnergy.yaml +++ b/config/STGODE/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STGODE seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,16 +17,19 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: history: 24 horizon: 24 input_dim: 137 num_features: 137 + num_nodes: 137 output_dim: 137 sigma1: 0.1 sigma2: 10 thres1: 0.6 thres2: 0.5 + train: batch_size: 16 debug: false @@ -39,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STID/AirQuality.yaml b/config/STID/AirQuality.yaml index f8abb05..f499161 100755 --- a/config/STID/AirQuality.yaml +++ b/config/STID/AirQuality.yaml @@ -13,7 +13,7 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 @@ -31,7 +31,7 @@ model: input_len: 24 node_dim: 32 num_layer: 3 - num_nodes: 35 + num_nodes: 12 output_dim: 1 output_len: 24 temp_dim_diw: 32 @@ -51,7 +51,7 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 diff --git a/config/STID/BJTaxi-InFlow.yaml b/config/STID/BJTaxi-InFlow.yaml index 57b8e7f..59e9501 100644 --- a/config/STID/BJTaxi-InFlow.yaml +++ b/config/STID/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STID seed: 2023 + data: batch_size: 64 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 day_of_week_size: 7 @@ -35,6 +37,7 @@ model: temp_dim_diw: 32 temp_dim_tid: 32 time_of_day_size: 288 + train: batch_size: 64 debug: true @@ -48,7 +51,7 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 diff --git a/config/STID/BJTaxi-OutFlow.yaml b/config/STID/BJTaxi-OutFlow.yaml index 4a10026..e2fdf43 100644 --- a/config/STID/BJTaxi-OutFlow.yaml +++ b/config/STID/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STID seed: 2023 + data: batch_size: 64 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 day_of_week_size: 7 @@ -35,6 +37,7 @@ model: temp_dim_diw: 32 temp_dim_tid: 32 time_of_day_size: 288 + train: batch_size: 64 debug: true @@ -48,7 +51,7 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 diff --git a/config/STID/BJTaxi_Inflow.yaml b/config/STID/BJTaxi_Inflow.yaml index d29f33f..d50ba22 100755 --- a/config/STID/BJTaxi_Inflow.yaml +++ b/config/STID/BJTaxi_Inflow.yaml @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 1 weight_decay: 0.0001 diff --git a/config/STID/BJTaxi_Outflow.yaml b/config/STID/BJTaxi_Outflow.yaml index 4c3b344..e2fdf43 100755 --- a/config/STID/BJTaxi_Outflow.yaml +++ b/config/STID/BJTaxi_Outflow.yaml @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 1 weight_decay: 0.0001 diff --git a/config/STID/METR-LA.yaml b/config/STID/METR-LA.yaml index d79894a..7ceb4f0 100755 --- a/config/STID/METR-LA.yaml +++ b/config/STID/METR-LA.yaml @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 1 weight_decay: 0.0001 diff --git a/config/STID/NYCBike-InFlow.yaml b/config/STID/NYCBike-InFlow.yaml index 81d392d..e509007 100644 --- a/config/STID/NYCBike-InFlow.yaml +++ b/config/STID/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STID seed: 2023 + data: batch_size: 64 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 day_of_week_size: 7 @@ -29,12 +31,13 @@ model: input_len: 24 node_dim: 32 num_layer: 3 - num_nodes: 1024 + num_nodes: 128 output_dim: 1 output_len: 24 temp_dim_diw: 32 temp_dim_tid: 32 time_of_day_size: 288 + train: batch_size: 64 debug: true @@ -48,7 +51,7 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 diff --git a/config/STID/NYCBike-OutFlow.yaml b/config/STID/NYCBike-OutFlow.yaml index dc305ce..155baf3 100644 --- a/config/STID/NYCBike-OutFlow.yaml +++ b/config/STID/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STID seed: 2023 + data: batch_size: 64 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: batch_size: 64 day_of_week_size: 7 @@ -29,12 +31,13 @@ model: input_len: 24 node_dim: 32 num_layer: 3 - num_nodes: 1024 + num_nodes: 128 output_dim: 1 output_len: 24 temp_dim_diw: 32 temp_dim_tid: 32 time_of_day_size: 288 + train: batch_size: 64 debug: true @@ -48,7 +51,7 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 diff --git a/config/STID/NYCBike_Inflow.yaml b/config/STID/NYCBike_Inflow.yaml index e014c20..e509007 100755 --- a/config/STID/NYCBike_Inflow.yaml +++ b/config/STID/NYCBike_Inflow.yaml @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 1 weight_decay: 0.0001 diff --git a/config/STID/NYCBike_Outflow.yaml b/config/STID/NYCBike_Outflow.yaml index 634600a..155baf3 100755 --- a/config/STID/NYCBike_Outflow.yaml +++ b/config/STID/NYCBike_Outflow.yaml @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 1 weight_decay: 0.0001 diff --git a/config/STID/PEMS-BAY.yaml b/config/STID/PEMS-BAY.yaml index 176c39f..561102d 100755 --- a/config/STID/PEMS-BAY.yaml +++ b/config/STID/PEMS-BAY.yaml @@ -51,7 +51,7 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: null mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 diff --git a/config/STID/PEMSD4.yaml b/config/STID/PEMSD4.yaml index ddfaf8f..84dee4d 100755 --- a/config/STID/PEMSD4.yaml +++ b/config/STID/PEMSD4.yaml @@ -50,11 +50,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 1 weight_decay: 0.0001 diff --git a/config/STID/SolarEnergy.yaml b/config/STID/SolarEnergy.yaml index 6fa3ad6..0d787c9 100755 --- a/config/STID/SolarEnergy.yaml +++ b/config/STID/SolarEnergy.yaml @@ -51,11 +51,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 1,50,80 lr_init: 0.002 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 1 weight_decay: 0.0001 diff --git a/config/STIDGCN/AirQuality.yaml b/config/STIDGCN/AirQuality.yaml index 116549e..5af3fd4 100644 --- a/config/STIDGCN/AirQuality.yaml +++ b/config/STIDGCN/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STIDGCN seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: channels: 32 dropout: 0.1 @@ -23,7 +25,9 @@ model: history: 24 horizon: 24 input_dim: 3 + num_nodes: 12 output_dim: 6 + train: batch_size: 16 debug: false @@ -37,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/BJTaxi-InFlow.yaml b/config/STIDGCN/BJTaxi-InFlow.yaml index 06c2aa5..26f8c52 100644 --- a/config/STIDGCN/BJTaxi-InFlow.yaml +++ b/config/STIDGCN/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STIDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: channels: 32 dropout: 0.1 @@ -23,7 +25,9 @@ model: history: 24 horizon: 24 input_dim: 3 + num_nodes: 1024 output_dim: 1 + train: batch_size: 32 debug: false @@ -37,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/BJTaxi-OutFlow.yaml b/config/STIDGCN/BJTaxi-OutFlow.yaml index dae5ec9..f09fa95 100644 --- a/config/STIDGCN/BJTaxi-OutFlow.yaml +++ b/config/STIDGCN/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STIDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: channels: 32 dropout: 0.1 @@ -23,7 +25,9 @@ model: history: 24 horizon: 24 input_dim: 3 + num_nodes: 1024 output_dim: 1 + train: batch_size: 32 debug: false @@ -37,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/METR-LA.yaml b/config/STIDGCN/METR-LA.yaml index fac77f5..1022a11 100644 --- a/config/STIDGCN/METR-LA.yaml +++ b/config/STIDGCN/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STIDGCN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: channels: 32 dropout: 0.1 @@ -23,7 +25,9 @@ model: history: 12 horizon: 12 input_dim: 3 + num_nodes: 207 output_dim: 1 + train: batch_size: 16 debug: false @@ -37,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/NYCBike-InFlow.yaml b/config/STIDGCN/NYCBike-InFlow.yaml index 1237d5c..df6f976 100644 --- a/config/STIDGCN/NYCBike-InFlow.yaml +++ b/config/STIDGCN/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STIDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: channels: 32 dropout: 0.1 @@ -23,7 +25,9 @@ model: history: 24 horizon: 24 input_dim: 3 + num_nodes: 128 output_dim: 1 + train: batch_size: 32 debug: false @@ -37,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/NYCBike-OutFlow.yaml b/config/STIDGCN/NYCBike-OutFlow.yaml index 3f95335..e7159a2 100644 --- a/config/STIDGCN/NYCBike-OutFlow.yaml +++ b/config/STIDGCN/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STIDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: channels: 32 dropout: 0.1 @@ -23,7 +25,9 @@ model: history: 24 horizon: 24 input_dim: 3 + num_nodes: 128 output_dim: 1 + train: batch_size: 32 debug: false @@ -37,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/PEMSD3.yaml b/config/STIDGCN/PEMSD3.yaml index d4548cc..a1fc024 100644 --- a/config/STIDGCN/PEMSD3.yaml +++ b/config/STIDGCN/PEMSD3.yaml @@ -25,6 +25,7 @@ model: history: 12 horizon: 12 input_dim: 3 + num_nodes: 358 output_dim: 1 train: @@ -40,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/PEMSD4.yaml b/config/STIDGCN/PEMSD4.yaml index 35c69be..edd4118 100644 --- a/config/STIDGCN/PEMSD4.yaml +++ b/config/STIDGCN/PEMSD4.yaml @@ -25,6 +25,7 @@ model: history: 12 horizon: 12 input_dim: 3 + num_nodes: 307 output_dim: 1 train: @@ -40,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/PEMSD7.yaml b/config/STIDGCN/PEMSD7.yaml index ba92d98..942ba1b 100644 --- a/config/STIDGCN/PEMSD7.yaml +++ b/config/STIDGCN/PEMSD7.yaml @@ -25,6 +25,7 @@ model: history: 12 horizon: 12 input_dim: 3 + num_nodes: 883 output_dim: 1 train: @@ -40,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/PEMSD8.yaml b/config/STIDGCN/PEMSD8.yaml index fca7310..071ab05 100644 --- a/config/STIDGCN/PEMSD8.yaml +++ b/config/STIDGCN/PEMSD8.yaml @@ -25,6 +25,7 @@ model: history: 12 horizon: 12 input_dim: 3 + num_nodes: 170 output_dim: 1 train: @@ -40,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STIDGCN/SolarEnergy.yaml b/config/STIDGCN/SolarEnergy.yaml index 243b0f8..e4d66ba 100644 --- a/config/STIDGCN/SolarEnergy.yaml +++ b/config/STIDGCN/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STIDGCN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: channels: 32 dropout: 0.1 @@ -23,7 +25,9 @@ model: history: 24 horizon: 24 input_dim: 3 + num_nodes: 137 output_dim: 137 + train: batch_size: 16 debug: false @@ -37,11 +41,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STMLP/AirQuality.yaml b/config/STMLP/AirQuality.yaml index 7166af6..d3bbdb9 100644 --- a/config/STMLP/AirQuality.yaml +++ b/config/STMLP/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STMLP seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: buildA_true: true conv_channels: 32 @@ -30,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 12 num_split: 1 output_dim: 6 output_window: 24 @@ -42,6 +45,7 @@ model: tanhalpha: 3 task_level: 0 use_curriculum_learning: true + train: batch_size: 16 debug: false @@ -55,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/BJTaxi-InFlow.yaml b/config/STMLP/BJTaxi-InFlow.yaml index 9b90e0d..1dacb06 100644 --- a/config/STMLP/BJTaxi-InFlow.yaml +++ b/config/STMLP/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STMLP seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: buildA_true: true conv_channels: 32 @@ -30,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 1024 num_split: 1 output_dim: 1 output_window: 24 @@ -42,6 +45,7 @@ model: tanhalpha: 3 task_level: 0 use_curriculum_learning: true + train: batch_size: 32 debug: false @@ -55,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/BJTaxi-OutFlow.yaml b/config/STMLP/BJTaxi-OutFlow.yaml index cf499e3..5b34c75 100644 --- a/config/STMLP/BJTaxi-OutFlow.yaml +++ b/config/STMLP/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STMLP seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: buildA_true: true conv_channels: 32 @@ -30,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 1024 num_split: 1 output_dim: 1 output_window: 24 @@ -42,6 +45,7 @@ model: tanhalpha: 3 task_level: 0 use_curriculum_learning: true + train: batch_size: 32 debug: false @@ -55,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/METR-LA.yaml b/config/STMLP/METR-LA.yaml index 5313959..426dec7 100644 --- a/config/STMLP/METR-LA.yaml +++ b/config/STMLP/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STMLP seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: buildA_true: true conv_channels: 32 @@ -30,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 207 num_split: 1 output_dim: 1 output_window: 12 @@ -42,6 +45,7 @@ model: tanhalpha: 3 task_level: 0 use_curriculum_learning: true + train: batch_size: 16 debug: false @@ -55,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/NYCBike-InFlow.yaml b/config/STMLP/NYCBike-InFlow.yaml index 053deab..ccbc983 100644 --- a/config/STMLP/NYCBike-InFlow.yaml +++ b/config/STMLP/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STMLP seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: buildA_true: true conv_channels: 32 @@ -30,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 128 num_split: 1 output_dim: 1 output_window: 24 @@ -42,6 +45,7 @@ model: tanhalpha: 3 task_level: 0 use_curriculum_learning: true + train: batch_size: 32 debug: false @@ -55,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/NYCBike-OutFlow.yaml b/config/STMLP/NYCBike-OutFlow.yaml index 0a920cc..a709a21 100644 --- a/config/STMLP/NYCBike-OutFlow.yaml +++ b/config/STMLP/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STMLP seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: buildA_true: true conv_channels: 32 @@ -30,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 128 num_split: 1 output_dim: 1 output_window: 24 @@ -42,6 +45,7 @@ model: tanhalpha: 3 task_level: 0 use_curriculum_learning: true + train: batch_size: 32 debug: false @@ -55,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/PEMSD3.yaml b/config/STMLP/PEMSD3.yaml index 55372aa..1bbaaad 100644 --- a/config/STMLP/PEMSD3.yaml +++ b/config/STMLP/PEMSD3.yaml @@ -32,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 358 num_split: 1 output_dim: 1 output_window: 12 @@ -58,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/PEMSD4.yaml b/config/STMLP/PEMSD4.yaml index 780bf77..f90156b 100644 --- a/config/STMLP/PEMSD4.yaml +++ b/config/STMLP/PEMSD4.yaml @@ -32,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 307 num_split: 1 output_dim: 1 output_window: 12 @@ -58,13 +59,12 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher: true teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/PEMSD7.yaml b/config/STMLP/PEMSD7.yaml index bae2ec9..6fb1de2 100644 --- a/config/STMLP/PEMSD7.yaml +++ b/config/STMLP/PEMSD7.yaml @@ -32,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 883 num_split: 1 output_dim: 1 output_window: 12 @@ -58,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/PEMSD8.yaml b/config/STMLP/PEMSD8.yaml index 6532996..5858ec5 100644 --- a/config/STMLP/PEMSD8.yaml +++ b/config/STMLP/PEMSD8.yaml @@ -32,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 170 num_split: 1 output_dim: 1 output_window: 12 @@ -58,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STMLP/SolarEnergy.yaml b/config/STMLP/SolarEnergy.yaml index ca6bc0d..627ba66 100644 --- a/config/STMLP/SolarEnergy.yaml +++ b/config/STMLP/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STMLP seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: buildA_true: true conv_channels: 32 @@ -30,6 +32,7 @@ model: layers: 3 model_type: stmlp node_dim: 40 + num_nodes: 137 num_split: 1 output_dim: 137 output_window: 24 @@ -42,6 +45,7 @@ model: tanhalpha: 3 task_level: 0 use_curriculum_learning: true + train: batch_size: 16 debug: false @@ -55,12 +59,11 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 teacher_stu: true weight_decay: 0 diff --git a/config/STSGCN/AirQuality.yaml b/config/STSGCN/AirQuality.yaml index 0b65da7..d9ccba1 100644 --- a/config/STSGCN/AirQuality.yaml +++ b/config/STSGCN/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STSGCN seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,36 +13,39 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 24 horizon: 24 input_dim: 6 + num_nodes: 12 out_layer_dim: 128 output_dim: 6 spatial_emb: true strides: 3 temporal_emb: true use_mask: true + train: batch_size: 16 debug: false @@ -55,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/BJTaxi-InFlow.yaml b/config/STSGCN/BJTaxi-InFlow.yaml index 074da90..ca137d5 100644 --- a/config/STSGCN/BJTaxi-InFlow.yaml +++ b/config/STSGCN/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STSGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,32 +17,35 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 24 horizon: 24 input_dim: 1 + num_nodes: 1024 out_layer_dim: 128 output_dim: 1 spatial_emb: true strides: 3 temporal_emb: true use_mask: true + train: batch_size: 32 debug: false @@ -55,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/BJTaxi-OutFlow.yaml b/config/STSGCN/BJTaxi-OutFlow.yaml index 1b395db..460d836 100644 --- a/config/STSGCN/BJTaxi-OutFlow.yaml +++ b/config/STSGCN/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STSGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,32 +17,35 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 24 horizon: 24 input_dim: 1 + num_nodes: 1024 out_layer_dim: 128 output_dim: 1 spatial_emb: true strides: 3 temporal_emb: true use_mask: true + train: batch_size: 32 debug: false @@ -55,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/METR-LA.yaml b/config/STSGCN/METR-LA.yaml index cd54e19..898a399 100644 --- a/config/STSGCN/METR-LA.yaml +++ b/config/STSGCN/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STSGCN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,32 +17,35 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 12 horizon: 12 input_dim: 1 + num_nodes: 207 out_layer_dim: 128 output_dim: 1 spatial_emb: true strides: 3 temporal_emb: true use_mask: true + train: batch_size: 16 debug: false @@ -55,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/NYCBike-InFlow.yaml b/config/STSGCN/NYCBike-InFlow.yaml index 02752c5..143e0fa 100644 --- a/config/STSGCN/NYCBike-InFlow.yaml +++ b/config/STSGCN/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STSGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,36 +13,39 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 24 horizon: 24 input_dim: 1 + num_nodes: 128 out_layer_dim: 128 output_dim: 1 spatial_emb: true strides: 3 temporal_emb: true use_mask: true + train: batch_size: 32 debug: false @@ -55,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/NYCBike-OutFlow.yaml b/config/STSGCN/NYCBike-OutFlow.yaml index 868d4d1..b78fcb0 100644 --- a/config/STSGCN/NYCBike-OutFlow.yaml +++ b/config/STSGCN/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STSGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,36 +13,39 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 24 horizon: 24 input_dim: 1 + num_nodes: 128 out_layer_dim: 128 output_dim: 1 spatial_emb: true strides: 3 temporal_emb: true use_mask: true + train: batch_size: 32 debug: false @@ -55,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/PEMSD3.yaml b/config/STSGCN/PEMSD3.yaml index 0b08995..38250e8 100755 --- a/config/STSGCN/PEMSD3.yaml +++ b/config/STSGCN/PEMSD3.yaml @@ -22,10 +22,23 @@ model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 - hidden_dims: [[64, 64, 64], [64, 64, 64], [64, 64, 64], [64, 64, 64]] + hidden_dims: + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 12 horizon: 12 input_dim: 1 + num_nodes: 358 out_layer_dim: 128 output_dim: 1 spatial_emb: true @@ -46,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/PEMSD4.yaml b/config/STSGCN/PEMSD4.yaml index b5ef9fe..bfa33c2 100755 --- a/config/STSGCN/PEMSD4.yaml +++ b/config/STSGCN/PEMSD4.yaml @@ -22,10 +22,23 @@ model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 - hidden_dims: [[64, 64, 64], [64, 64, 64], [64, 64, 64], [64, 64, 64]] + hidden_dims: + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 12 horizon: 12 input_dim: 1 + num_nodes: 307 out_layer_dim: 128 output_dim: 1 spatial_emb: true @@ -46,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/PEMSD7.yaml b/config/STSGCN/PEMSD7.yaml index c13cde9..e1ef225 100755 --- a/config/STSGCN/PEMSD7.yaml +++ b/config/STSGCN/PEMSD7.yaml @@ -22,10 +22,23 @@ model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 - hidden_dims: [[64, 64, 64], [64, 64, 64], [64, 64, 64], [64, 64, 64]] + hidden_dims: + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 12 horizon: 12 input_dim: 1 + num_nodes: 883 out_layer_dim: 128 output_dim: 1 spatial_emb: true @@ -46,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/PEMSD8.yaml b/config/STSGCN/PEMSD8.yaml index df35209..183646d 100755 --- a/config/STSGCN/PEMSD8.yaml +++ b/config/STSGCN/PEMSD8.yaml @@ -22,10 +22,23 @@ model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 - hidden_dims: [[64, 64, 64], [64, 64, 64], [64, 64, 64], [64, 64, 64]] + hidden_dims: + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 12 horizon: 12 input_dim: 1 + num_nodes: 170 out_layer_dim: 128 output_dim: 1 spatial_emb: true @@ -46,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/STSGCN/SolarEnergy.yaml b/config/STSGCN/SolarEnergy.yaml index 9baa783..3f25c2a 100644 --- a/config/STSGCN/SolarEnergy.yaml +++ b/config/STSGCN/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: STSGCN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,32 +17,35 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: activation: GLU construct_type: connectivity first_layer_embedding_size: 64 hidden_dims: - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 - - - 64 - - 64 - - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 + - - 64 + - 64 + - 64 history: 24 horizon: 24 input_dim: 137 + num_nodes: 137 out_layer_dim: 128 output_dim: 137 spatial_emb: true strides: 3 temporal_emb: true use_mask: true + train: batch_size: 16 debug: false @@ -55,11 +59,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/AirQuality.yaml b/config/ST_SSL/AirQuality.yaml index a1ecc1d..e2459b3 100644 --- a/config/ST_SSL/AirQuality.yaml +++ b/config/ST_SSL/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ST_SSL seed: 2023 + data: batch_size: 16 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: d_model: 64 dropout: 0.1 @@ -24,10 +26,12 @@ model: input_dim: 6 n_his: 24 nmb_prototype: 10 + num_nodes: 12 output_dim: 6 percent: 0.1 shm_temp: 0.1 yita: 0.5 + train: batch_size: 16 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/BJTaxi-InFlow.yaml b/config/ST_SSL/BJTaxi-InFlow.yaml index 4dbf256..6da077f 100644 --- a/config/ST_SSL/BJTaxi-InFlow.yaml +++ b/config/ST_SSL/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ST_SSL seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_model: 64 dropout: 0.1 @@ -24,10 +26,12 @@ model: input_dim: 1 n_his: 24 nmb_prototype: 10 + num_nodes: 1024 output_dim: 1 percent: 0.1 shm_temp: 0.1 yita: 0.5 + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/BJTaxi-OutFlow.yaml b/config/ST_SSL/BJTaxi-OutFlow.yaml index 6801117..969be92 100644 --- a/config/ST_SSL/BJTaxi-OutFlow.yaml +++ b/config/ST_SSL/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ST_SSL seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_model: 64 dropout: 0.1 @@ -24,10 +26,12 @@ model: input_dim: 1 n_his: 24 nmb_prototype: 10 + num_nodes: 1024 output_dim: 1 percent: 0.1 shm_temp: 0.1 yita: 0.5 + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/METR-LA.yaml b/config/ST_SSL/METR-LA.yaml index d80ccb9..7805bc4 100644 --- a/config/ST_SSL/METR-LA.yaml +++ b/config/ST_SSL/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ST_SSL seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: d_model: 64 dropout: 0.1 @@ -24,10 +26,12 @@ model: input_dim: 1 n_his: 12 nmb_prototype: 10 + num_nodes: 207 output_dim: 1 percent: 0.1 shm_temp: 0.1 yita: 0.5 + train: batch_size: 16 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/NYCBike-InFlow.yaml b/config/ST_SSL/NYCBike-InFlow.yaml index 3283cb6..af85c3a 100644 --- a/config/ST_SSL/NYCBike-InFlow.yaml +++ b/config/ST_SSL/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ST_SSL seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_model: 64 dropout: 0.1 @@ -24,10 +26,12 @@ model: input_dim: 1 n_his: 24 nmb_prototype: 10 + num_nodes: 128 output_dim: 1 percent: 0.1 shm_temp: 0.1 yita: 0.5 + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/NYCBike-OutFlow.yaml b/config/ST_SSL/NYCBike-OutFlow.yaml index 3a3e06f..e3b0c3c 100644 --- a/config/ST_SSL/NYCBike-OutFlow.yaml +++ b/config/ST_SSL/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ST_SSL seed: 2023 + data: batch_size: 32 column_wise: false @@ -12,10 +13,11 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 1024 + num_nodes: 128 steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: d_model: 64 dropout: 0.1 @@ -24,10 +26,12 @@ model: input_dim: 1 n_his: 24 nmb_prototype: 10 + num_nodes: 128 output_dim: 1 percent: 0.1 shm_temp: 0.1 yita: 0.5 + train: batch_size: 32 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/PEMSD3.yaml b/config/ST_SSL/PEMSD3.yaml index 70dc619..c9c933c 100644 --- a/config/ST_SSL/PEMSD3.yaml +++ b/config/ST_SSL/PEMSD3.yaml @@ -26,6 +26,7 @@ model: input_dim: 1 n_his: 12 nmb_prototype: 10 + num_nodes: 358 output_dim: 1 percent: 0.1 shm_temp: 0.1 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/PEMSD4.yaml b/config/ST_SSL/PEMSD4.yaml index 2aab8c8..7a53a20 100644 --- a/config/ST_SSL/PEMSD4.yaml +++ b/config/ST_SSL/PEMSD4.yaml @@ -26,6 +26,7 @@ model: input_dim: 1 n_his: 12 nmb_prototype: 10 + num_nodes: 307 output_dim: 1 percent: 0.1 shm_temp: 0.1 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/PEMSD7.yaml b/config/ST_SSL/PEMSD7.yaml index 4ec9986..75019a3 100644 --- a/config/ST_SSL/PEMSD7.yaml +++ b/config/ST_SSL/PEMSD7.yaml @@ -26,6 +26,7 @@ model: input_dim: 1 n_his: 12 nmb_prototype: 10 + num_nodes: 883 output_dim: 1 percent: 0.1 shm_temp: 0.1 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/PEMSD8.yaml b/config/ST_SSL/PEMSD8.yaml index 1129639..9aeb16f 100644 --- a/config/ST_SSL/PEMSD8.yaml +++ b/config/ST_SSL/PEMSD8.yaml @@ -26,6 +26,7 @@ model: input_dim: 1 n_his: 12 nmb_prototype: 10 + num_nodes: 170 output_dim: 1 percent: 0.1 shm_temp: 0.1 @@ -44,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/ST_SSL/SolarEnergy.yaml b/config/ST_SSL/SolarEnergy.yaml index cbe6d71..2752c44 100644 --- a/config/ST_SSL/SolarEnergy.yaml +++ b/config/ST_SSL/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: ST_SSL seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: d_model: 64 dropout: 0.1 @@ -24,10 +26,12 @@ model: input_dim: 137 n_his: 24 nmb_prototype: 10 + num_nodes: 137 output_dim: 137 percent: 0.1 shm_temp: 0.1 yita: 0.5 + train: batch_size: 16 debug: false @@ -41,11 +45,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: '' + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 137 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TCN/AirQuality.yaml b/config/TCN/AirQuality.yaml index 3b19a78..c04eeac 100644 --- a/config/TCN/AirQuality.yaml +++ b/config/TCN/AirQuality.yaml @@ -13,7 +13,7 @@ data: input_dim: 6 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 @@ -21,10 +21,14 @@ data: model: batch_size: 16 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 6 kernel_size: 3 num_layers: 3 + num_nodes: 12 output_dim: 6 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 6 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/TCN/BJTaxi-InFlow.yaml b/config/TCN/BJTaxi-InFlow.yaml index 68cf5e3..c49b1dc 100644 --- a/config/TCN/BJTaxi-InFlow.yaml +++ b/config/TCN/BJTaxi-InFlow.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 32 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 1024 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/TCN/BJTaxi-OutFlow.yaml b/config/TCN/BJTaxi-OutFlow.yaml index 377bb5b..077c9a3 100644 --- a/config/TCN/BJTaxi-OutFlow.yaml +++ b/config/TCN/BJTaxi-OutFlow.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 32 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 1024 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/TCN/METR-LA.yaml b/config/TCN/METR-LA.yaml index 10fc585..a588114 100644 --- a/config/TCN/METR-LA.yaml +++ b/config/TCN/METR-LA.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 16 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 207 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/TCN/NYCBike-InFlow.yaml b/config/TCN/NYCBike-InFlow.yaml index cd5242c..f5005d0 100644 --- a/config/TCN/NYCBike-InFlow.yaml +++ b/config/TCN/NYCBike-InFlow.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 32 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 128 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/TCN/NYCBike-OutFlow.yaml b/config/TCN/NYCBike-OutFlow.yaml index 761431b..0d06c3b 100644 --- a/config/TCN/NYCBike-OutFlow.yaml +++ b/config/TCN/NYCBike-OutFlow.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 32 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 128 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/TCN/PEMSD3.yaml b/config/TCN/PEMSD3.yaml index 47a59f2..00396e0 100755 --- a/config/TCN/PEMSD3.yaml +++ b/config/TCN/PEMSD3.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 64 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 358 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TCN/PEMSD4.yaml b/config/TCN/PEMSD4.yaml index 810859f..552df9e 100755 --- a/config/TCN/PEMSD4.yaml +++ b/config/TCN/PEMSD4.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 64 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 307 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TCN/PEMSD7.yaml b/config/TCN/PEMSD7.yaml index 6436803..ca414f2 100755 --- a/config/TCN/PEMSD7.yaml +++ b/config/TCN/PEMSD7.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 64 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 883 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TCN/PEMSD8.yaml b/config/TCN/PEMSD8.yaml index d47fdc5..e0f4761 100755 --- a/config/TCN/PEMSD8.yaml +++ b/config/TCN/PEMSD8.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 64 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 170 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TCN/SolarEnergy.yaml b/config/TCN/SolarEnergy.yaml index d620185..b12c1bf 100644 --- a/config/TCN/SolarEnergy.yaml +++ b/config/TCN/SolarEnergy.yaml @@ -21,10 +21,14 @@ data: model: batch_size: 64 dropout: 0.2 - hidden_channels: [32, 64, 32] + hidden_channels: + - 32 + - 64 + - 32 input_dim: 1 kernel_size: 3 num_layers: 3 + num_nodes: 137 output_dim: 1 train: @@ -40,11 +44,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.001 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 - weight_decay: 0 \ No newline at end of file + weight_decay: 0 diff --git a/config/TWDGCN/AirQuality.yaml b/config/TWDGCN/AirQuality.yaml index 97f31a1..6cef32e 100644 --- a/config/TWDGCN/AirQuality.yaml +++ b/config/TWDGCN/AirQuality.yaml @@ -4,6 +4,7 @@ basic: mode: train model: TWDGCN seed: 2023 + data: batch_size: 64 column_wise: false @@ -12,21 +13,23 @@ data: input_dim: 1 lag: 24 normalizer: std - num_nodes: 35 + num_nodes: 12 steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 12 horizon: 24 input_dim: 1 num_layers: 1 - num_nodes: 35 + num_nodes: 12 output_dim: 1 rnn_units: 64 use_day: true use_week: false + train: batch_size: 64 debug: false @@ -46,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/BJTaxi-InFlow.yaml b/config/TWDGCN/BJTaxi-InFlow.yaml index cf543c8..63b28d1 100644 --- a/config/TWDGCN/BJTaxi-InFlow.yaml +++ b/config/TWDGCN/BJTaxi-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: TWDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 12 @@ -27,6 +29,7 @@ model: rnn_units: 64 use_day: true use_week: false + train: batch_size: 32 debug: false @@ -46,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/BJTaxi-OutFlow.yaml b/config/TWDGCN/BJTaxi-OutFlow.yaml index a9ff5f9..feb1636 100644 --- a/config/TWDGCN/BJTaxi-OutFlow.yaml +++ b/config/TWDGCN/BJTaxi-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: TWDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 12 @@ -27,6 +29,7 @@ model: rnn_units: 64 use_day: true use_week: false + train: batch_size: 32 debug: false @@ -46,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/Hainan.yaml b/config/TWDGCN/Hainan.yaml index 7774f92..058ca11 100755 --- a/config/TWDGCN/Hainan.yaml +++ b/config/TWDGCN/Hainan.yaml @@ -13,7 +13,7 @@ data: input_dim: 1 lag: 12 normalizer: std - num_nodes: 13 + num_nodes: 200 steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 @@ -25,7 +25,7 @@ model: horizon: 12 input_dim: 1 num_layers: 1 - num_nodes: 13 + num_nodes: 200 output_dim: 1 rnn_units: 32 use_day: true @@ -48,7 +48,7 @@ train: - 40 - 70 lr_init: 0.003 - mae_thresh: + mae_thresh: null mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 diff --git a/config/TWDGCN/METR-LA.yaml b/config/TWDGCN/METR-LA.yaml index 0788a9d..29015a7 100644 --- a/config/TWDGCN/METR-LA.yaml +++ b/config/TWDGCN/METR-LA.yaml @@ -4,6 +4,7 @@ basic: mode: train model: TWDGCN seed: 2023 + data: batch_size: 16 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 288 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 12 @@ -27,6 +29,7 @@ model: rnn_units: 64 use_day: true use_week: false + train: batch_size: 16 debug: false @@ -46,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/NYCBike-InFlow.yaml b/config/TWDGCN/NYCBike-InFlow.yaml index af27b83..0ca0c1d 100644 --- a/config/TWDGCN/NYCBike-InFlow.yaml +++ b/config/TWDGCN/NYCBike-InFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: TWDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 12 @@ -27,6 +29,7 @@ model: rnn_units: 64 use_day: true use_week: false + train: batch_size: 32 debug: false @@ -46,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/NYCBike-OutFlow.yaml b/config/TWDGCN/NYCBike-OutFlow.yaml index 2b509a1..7226490 100644 --- a/config/TWDGCN/NYCBike-OutFlow.yaml +++ b/config/TWDGCN/NYCBike-OutFlow.yaml @@ -4,6 +4,7 @@ basic: mode: train model: TWDGCN seed: 2023 + data: batch_size: 32 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 48 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 12 @@ -27,6 +29,7 @@ model: rnn_units: 64 use_day: true use_week: false + train: batch_size: 32 debug: false @@ -46,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/PEMSD3.yaml b/config/TWDGCN/PEMSD3.yaml index 196970e..04a0311 100755 --- a/config/TWDGCN/PEMSD3.yaml +++ b/config/TWDGCN/PEMSD3.yaml @@ -43,11 +43,10 @@ train: lr_decay_rate: 0.3 lr_decay_step: 5,20,40,70 lr_init: 0.003 - mae_thresh: + mae_thresh: 0.0 mape_thresh: 0.0 max_grad_norm: 5 output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/PEMSD4.yaml b/config/TWDGCN/PEMSD4.yaml index c8e14a6..131c972 100755 --- a/config/TWDGCN/PEMSD4.yaml +++ b/config/TWDGCN/PEMSD4.yaml @@ -24,7 +24,7 @@ model: horizon: 12 input_dim: 1 num_layers: 1 - num_nodes: 307 + num_nodes: 307 output_dim: 1 rnn_units: 64 use_day: true @@ -49,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/PEMSD7(L).yaml b/config/TWDGCN/PEMSD7(L).yaml index f83a8ad..1cb5f73 100755 --- a/config/TWDGCN/PEMSD7(L).yaml +++ b/config/TWDGCN/PEMSD7(L).yaml @@ -51,5 +51,4 @@ train: output_dim: 1 plot: true real_value: true - seed: 12 weight_decay: 0 diff --git a/config/TWDGCN/PEMSD7(M).yaml b/config/TWDGCN/PEMSD7(M).yaml index c1bcd2f..19fc9a8 100755 --- a/config/TWDGCN/PEMSD7(M).yaml +++ b/config/TWDGCN/PEMSD7(M).yaml @@ -51,5 +51,4 @@ train: output_dim: 1 plot: true real_value: true - seed: 12 weight_decay: 0 diff --git a/config/TWDGCN/PEMSD7.yaml b/config/TWDGCN/PEMSD7.yaml index 2f79918..6861724 100755 --- a/config/TWDGCN/PEMSD7.yaml +++ b/config/TWDGCN/PEMSD7.yaml @@ -53,5 +53,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/config/TWDGCN/PEMSD8.yaml b/config/TWDGCN/PEMSD8.yaml index 6dac03a..37cceea 100755 --- a/config/TWDGCN/PEMSD8.yaml +++ b/config/TWDGCN/PEMSD8.yaml @@ -49,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 3407 weight_decay: 0 diff --git a/config/TWDGCN/SolarEnergy.yaml b/config/TWDGCN/SolarEnergy.yaml index 859116a..da5d9db 100644 --- a/config/TWDGCN/SolarEnergy.yaml +++ b/config/TWDGCN/SolarEnergy.yaml @@ -4,6 +4,7 @@ basic: mode: train model: TWDGCN seed: 2023 + data: batch_size: 64 column_wise: false @@ -16,6 +17,7 @@ data: steps_per_day: 24 test_ratio: 0.2 val_ratio: 0.2 + model: cheb_order: 2 embed_dim: 12 @@ -27,6 +29,7 @@ model: rnn_units: 64 use_day: true use_week: false + train: batch_size: 64 debug: false @@ -46,5 +49,4 @@ train: output_dim: 1 plot: false real_value: true - seed: 10 weight_decay: 0 diff --git a/test_configs.py b/test_configs.py deleted file mode 100644 index 4fb4bbe..0000000 --- a/test_configs.py +++ /dev/null @@ -1,124 +0,0 @@ -import os -import subprocess -import yaml -import time - -# 配置路径 -CONFIG_DIR = "/user/czzhangheng/code/TrafficWheel/config" -RUN_SCRIPT = "/user/czzhangheng/code/TrafficWheel/run.py" -RESULTS_FILE = "/user/czzhangheng/code/TrafficWheel/test_results.txt" - -# 记录测试结果的字典 -results = { - "passed": [], - "failed": [], - "error": [] -} - -# 遍历所有yaml文件 -def find_all_yaml_files(directory): - yaml_files = [] - for root, dirs, files in os.walk(directory): - for file in files: - if file.endswith(".yaml") and not file.startswith("BJTaxi"): - yaml_files.append(os.path.join(root, file)) - return yaml_files - -# 测试单个yaml文件 -def test_yaml_file(yaml_path): - print(f"\n=== Testing {yaml_path} ===") - - # 检查文件是否存在 - if not os.path.exists(yaml_path): - print(f"File not found: {yaml_path}") - return "error", f"File not found: {yaml_path}" - - # 运行测试命令 - command = ["python", RUN_SCRIPT, "--config", yaml_path] - try: - result = subprocess.run( - command, - capture_output=True, - text=True, - timeout=600 # 10分钟超时 - ) - - # 分析结果 - if result.returncode == 0: - if "Test passed" in result.stdout: - print(f"✓ PASSED: {yaml_path}") - return "passed", result.stdout.strip() - else: - print(f"✗ FAILED: {yaml_path}") - return "failed", result.stdout.strip() + "\n" + result.stderr.strip() - else: - print(f"✗ ERROR: {yaml_path}") - return "error", result.stdout.strip() + "\n" + result.stderr.strip() - except subprocess.TimeoutExpired: - print(f"✗ TIMEOUT: {yaml_path}") - return "error", "Timeout after 10 minutes" - except Exception as e: - print(f"✗ EXCEPTION: {yaml_path}") - return "error", str(e) - -# 生成测试报告 -def generate_report(results): - total = len(results["passed"]) + len(results["failed"]) + len(results["error"]) - - report = f"""# 测试报告 - -## 测试概述 -- 测试时间: {time.strftime('%Y-%m-%d %H:%M:%S')} -- 总测试文件数: {total} -- 通过: {len(results['passed'])} -- 失败: {len(results['failed'])} -- 错误: {len(results['error'])} - -## 通过的配置文件 -""" - - for file_path, output in results["passed"]: - report += f"- ✅ {file_path}\n" - - report += "\n## 失败的配置文件\n" - for file_path, output in results["failed"]: - report += f"- ❌ {file_path}\n" - - report += "\n## 出错的配置文件\n" - for file_path, output in results["error"]: - report += f"- ⚠️ {file_path}\n" - - report += "\n## 详细输出\n" - - for status, files in results.items(): - report += f"\n### {status.upper()}\n\n" - for file_path, output in files: - report += f"#### {file_path}\n\n```\n{output}\n```\n\n" - - return report - -# 主函数 -def main(): - # 找到所有符合条件的yaml文件 - yaml_files = find_all_yaml_files(CONFIG_DIR) - print(f"Found {len(yaml_files)} yaml files to test") - - # 测试每个文件 - for yaml_file in yaml_files: - status, output = test_yaml_file(yaml_file) - results[status].append((yaml_file, output)) - - # 生成并保存报告 - report = generate_report(results) - with open(RESULTS_FILE, "w") as f: - f.write(report) - - print(f"\n=== Test Results ===") - print(f"Total: {len(yaml_files)}") - print(f"Passed: {len(results['passed'])}") - print(f"Failed: {len(results['failed'])}") - print(f"Error: {len(results['error'])}") - print(f"Report saved to: {RESULTS_FILE}") - -if __name__ == "__main__": - main() \ No newline at end of file