From f3480fccdc7b463e9ff4415212c151a81ba99482 Mon Sep 17 00:00:00 2001
From: czzhangheng <cz189@qq.com>
Date: Mon, 10 Nov 2025 09:34:52 +0800
Subject: [PATCH] =?UTF-8?q?STID=E9=85=8D=E7=BD=AE=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config/STID/PEMS-BAY.yaml | 66 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100755 config/STID/PEMS-BAY.yaml

diff --git a/config/STID/PEMS-BAY.yaml b/config/STID/PEMS-BAY.yaml
new file mode 100755
index 0000000..0a10a68
--- /dev/null
+++ b/config/STID/PEMS-BAY.yaml
@@ -0,0 +1,66 @@
+basic:
+  dataset: "PEMS-BAY"
+  mode: "train"
+  device: "cuda:0"
+  model: "STID"
+
+data:
+  num_nodes: 325
+  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: 288
+  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: 325
+  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: 300
+  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.0
+  log_step: 200
+  plot: False