search

QData / spacetimeformer

Multivariate Time Series Forecasting with efficient Transformers. Code for the paper "Long-Range Transformers for Dynamic Spatiotemporal Forecasting."
https://arxiv.org/abs/2109.12218
MIT License
808 stars 191 forks source link

Datasets #7

Closed LeftAttention closed 3 years ago

LeftAttention commented 3 years ago

Could you please share precip and asos datasets? I am very curious to test this model on these datasets.

jakegrigsby commented 3 years ago

The asos dataset should already be provided (?). It's data/temperature-v1.csv and the training script should find it. You should be able to get the precip dataset from this google drive link. Place the usgs_precipitation folder in spacetimeformer/data/.

Fair warning: the Precip task is pretty difficult and we did not include the results in the paper; I thought it was interesting and already had all the code for it so I kept it in this release. This could be because of a bug in the dataset loading, or possibly because the "radar" configuration is too spaced out (or maybe even too close?). Please let me know if you find anything!

LeftAttention commented 3 years ago

Thank you so much.