It is pytorch and keras implementations of EA-LSTM: Evolutionary Attention-based LSTM for Time Series Prediction.
- pandas
- numpy
- tqdm
- scikit-learn
- pytorch >= 1.3.0
- tensorflow >= 2.1.0
- You can simply run code by
$ python main.py --data pollution.csv
- 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.
For keras implementation, I refer to author's original code.