项目作者: bzantium

项目描述 :
EA-LSTM: Evolutionary attention-based LSTM for time series prediction
高级语言: Python
项目地址: git://github.com/bzantium/EA-LSTM.git
创建时间: 2020-01-04T05:27:54Z
项目社区:https://github.com/bzantium/EA-LSTM

开源协议:BSD 3-Clause "New" or "Revised" License

下载


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
    1. $ 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.