spacetimeformer icon indicating copy to clipboard operation
spacetimeformer copied to clipboard

Where to find used data?

Open DenisUllmann opened this issue 2 years ago • 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

DenisUllmann avatar Mar 07 '23 14:03 DenisUllmann

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

szdrnja avatar Sep 15 '23 13:09 szdrnja