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()