From 726e6dee3c254d5ee0162cbe74169644171edf94 Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Sat, 15 Nov 2025 19:36:48 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=9D=E6=8C=81=E5=8E=9F=E5=A7=8B=E9=85=8D?= =?UTF-8?q?=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 2 +- prepare_pems_bay.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 5e54f9b..45e3ed4 100644 --- a/README.md +++ b/README.md @@ -33,5 +33,5 @@ mkdir log 开跑 ```python -python run.py --root_path datasets --data_path PEMS-BAY --device cuda:0 --seq_len 12 --pred_len 12 +python run.py --root_path datasets --data_path PEMS-BAY --device cuda:0 --seq_len 24 --pred_len 24 ``` \ No newline at end of file diff --git a/prepare_pems_bay.py b/prepare_pems_bay.py index 0cd1758..bd6a58d 100644 --- a/prepare_pems_bay.py +++ b/prepare_pems_bay.py @@ -46,8 +46,8 @@ def split_by_ratio(x: np.ndarray, y: np.ndarray, ratios): def main(): parser = argparse.ArgumentParser(description="Prepare PEMS-BAY to train/val/test .npz") parser.add_argument("--dataset_dir", type=str, default='./datasets/PEMS-BAY', help="Path to datasets/PEMS-BAY directory") - parser.add_argument("--seq_len", type=int, default=12) - parser.add_argument("--pred_len", type=int, default=12) + parser.add_argument("--seq_len", type=int, default=24) + parser.add_argument("--pred_len", type=int, default=24) parser.add_argument("--speed_channel_only", action="store_true", help="Use only the first channel (speed)") args = parser.parse_args()