spacetimeformer
spacetimeformer copied to clipboard
Where to find used data?
Please let me know where I can find eg. PEMS-BAY data in the format ready for your model: /pems_bay/train.npz (and /val.npz and /test.npz) Or how to generate properly this data from any PEMS-BAY data available online.
Thanks in advance
Use this https://github.com/liyaguang/DCRNN repo, specifically
python -m scripts.generate_training_data --output_dir=data/PEMS-BAY --traffic_df_filename=data/pems-bay.h5
and set add_day_in_week to True because this code expects those features to be present. https://github.com/liyaguang/DCRNN/blob/master/scripts/generate_training_data.py#L72