/STA-LSTM

Spatio-temporal attention LSTM model for Spatio-temporal series problems

Primary LanguagePython

STA-LSTM

Spatio-temporal attention LSTM model for spatio-temporal series problems.

The code is a simple [Pytorch] version. We think it can help you to understand our paper better as it has all the details.

Thanks for your attention! Good luck in your research!

Don't forget to add our paper to your reference.

Yukai Ding, Yuelong Zhu, Jun Feng, et al. Interpretable spatio-temporal attention LSTM model for flood forecasting. Neurocomputing 2020(403): 348-359.

@article{DBLP:journals/ijon/DingZFZC20, author = {Yukai Ding and Yuelong Zhu and Jun Feng and Pengcheng Zhang and Zirun Cheng}, title = {Interpretable spatio-temporal attention {LSTM} model for flood forecasting}, journal = {Neurocomputing}, volume = {403}, pages = {348--359}, year = {2020}, url = {https://doi.org/10.1016/j.neucom.2020.04.110}, doi = {10.1016/j.neucom.2020.04.110}, timestamp = {Mon, 26 Oct 2020 08:37:31 +0100}, biburl = {https://dblp.org/rec/journals/ijon/DingZFZC20.bib}, bibsource = {dblp computer science bibliography, https://dblp.org} }