/EA-LSTM

EA-LSTM: Evolutionary attention-based LSTM for time series prediction

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

EA-LSTM

It is pytorch and keras implementations of EA-LSTM: Evolutionary Attention-based LSTM for Time Series Prediction.

Requirements

  • pandas
  • numpy
  • tqdm
  • scikit-learn
  • pytorch >= 1.3.0
  • tensorflow >= 2.1.0

Train and test

  • You can simply run code by
$ python main.py --data pollution.csv

Discussion

  • With my implementations, I failed to reproduce the results provided in the paper for PM2.5 dataset. if there's mistake or you have comment, please let me know to improve the code.

Reference

For keras implementation, I refer to author's original code.