QData/spacetimeformer

Where to find used data?

DenisUllmann opened this issue · 1 comments

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