czzhangheng
/
REPST
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

保持原始配置

This commit is contained in:
czzhangheng 2025-11-15 19:36:48 +08:00
parent 055fef5046
commit 726e6dee3c
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -33,5 +33,5 @@ mkdir log
开跑 开跑
```python ```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
``` ```

4
prepare_pems_bay.py
View File

@ -46,8 +46,8 @@ def split_by_ratio(x: np.ndarray, y: np.ndarray, ratios):
def main(): def main():
parser = argparse.ArgumentParser(description="Prepare PEMS-BAY to train/val/test .npz") 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("--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("--seq_len", type=int, default=24)
parser.add_argument("--pred_len", type=int, default=12) 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)") parser.add_argument("--speed_channel_only", action="store_true", help="Use only the first channel (speed)")
args = parser.parse_args() args = parser.parse_args()