diff --git a/.vscode/launch.json b/.vscode/launch.json index f8690b7..c9f73d8 100644 --- a/.vscode/launch.json +++ b/.vscode/launch.json @@ -60,6 +60,14 @@ "console": "integratedTerminal", "args": "--config ./config/STID/NYCBike_Outflow.yaml" }, + { + "name": "STID-SolarEnergy", + "type": "debugpy", + "request": "launch", + "program": "run.py", + "console": "integratedTerminal", + "args": "--config ./config/STID/SolarEnergy.yaml" + }, { "name": "REPST-BJTaxi-InFlow", "type": "debugpy", diff --git a/config/STID/AirQuality.yaml b/config/STID/AirQuality.yaml new file mode 100755 index 0000000..a9b4959 --- /dev/null +++ b/config/STID/AirQuality.yaml @@ -0,0 +1,66 @@ +basic: + dataset: "AirQuality" + mode: "train" + device: "cuda:0" + model: "STID" + seed: 2023 + +data: + num_nodes: 35 + lag: 24 + horizon: 24 + val_ratio: 0.2 + test_ratio: 0.2 + tod: False + normalizer: std + column_wise: False + default_graph: True + add_time_in_day: True + add_day_in_week: True + steps_per_day: 48 + days_per_week: 7 + input_dim: 1 + output_dim: 1 + batch_size: 64 + +model: + input_dim: 3 + output_dim: 1 + history: 24 + horizon: 24 + num_nodes: 35 + input_len: 24 + embed_dim: 32 + output_len: 24 + num_layer: 3 + if_node: True + node_dim: 32 + if_T_i_D: True + if_D_i_W: True + temp_dim_tid: 32 + temp_dim_diw: 32 + time_of_day_size: 288 + day_of_week_size: 7 + batch_size: 64 + + +train: + loss_func: mae + batch_size: 64 + epochs: 100 + lr_init: 0.002 + weight_decay: 0.0001 + lr_decay: False + lr_decay_rate: 0.3 + lr_decay_step: "1,50,80" + early_stop: True + early_stop_patience: 15 + grad_norm: False + max_grad_norm: 5 + real_value: True + debug: true + output_dim: 1 + mae_thresh: null + mape_thresh: 0.001 + log_step: 200 + plot: False diff --git a/config/STID/SolarEnergy.yaml b/config/STID/SolarEnergy.yaml new file mode 100755 index 0000000..2523d45 --- /dev/null +++ b/config/STID/SolarEnergy.yaml @@ -0,0 +1,67 @@ +basic: + dataset: "SolarEnergy" + mode: "train" + device: "cuda:0" + model: "STID" + seed: 2023 + +data: + num_nodes: 137 + lag: 24 + horizon: 24 + val_ratio: 0.2 + test_ratio: 0.2 + tod: False + normalizer: std + column_wise: False + default_graph: True + add_time_in_day: True + add_day_in_week: True + steps_per_day: 48 + days_per_week: 7 + input_dim: 1 + output_dim: 1 + batch_size: 64 + +model: + input_dim: 3 + output_dim: 1 + history: 24 + horizon: 24 + num_nodes: 137 + input_len: 24 + embed_dim: 32 + output_len: 24 + num_layer: 3 + if_node: True + node_dim: 32 + if_T_i_D: True + if_D_i_W: True + temp_dim_tid: 32 + temp_dim_diw: 32 + time_of_day_size: 288 + day_of_week_size: 7 + batch_size: 64 + + +train: + loss_func: mae + seed: 1 + batch_size: 64 + epochs: 100 + lr_init: 0.002 + weight_decay: 0.0001 + lr_decay: False + lr_decay_rate: 0.3 + lr_decay_step: "1,50,80" + early_stop: True + early_stop_patience: 15 + grad_norm: False + max_grad_norm: 5 + real_value: True + debug: true + output_dim: 1 + mae_thresh: null + mape_thresh: 0.001 + log_step: 200 + plot: False diff --git a/requirements.txt b/requirements.txt index 6199d59..c964fff 100755 --- a/requirements.txt +++ b/requirements.txt @@ -12,4 +12,5 @@ notebook torchcde einops transformers -py7zr \ No newline at end of file +py7zr +requests \ No newline at end of file