Skip to content

Latest commit

 

History

History
52 lines (47 loc) · 1.89 KB

README.md

File metadata and controls

52 lines (47 loc) · 1.89 KB

Time series prediction

LICENSE
This repository implements the common methods of time series prediction, especially deep learning methods in TensorFlow2. It's welcomed to contribute if you have any better idea, just create a PR. If any question, feel free to open an issue.

Ongoing project, I will continue to improve this, so you might want to watch/star this repo to revisit.

Usage

  1. Install the required library
$ pip install -r requirements.txt
  1. Download the data, if necessary
$ sh ./data/download_passenger.sh
  1. Train the model
    set custom_model_params if you want (refer to params in ./tfts/models/*.py), and pay attention to feature engineering.
$ cd examples
$ python run_train.py --use_model seq2seq
$ cd ..
$ tensorboard --logdir=./data/logs
  1. Predict new data
$ cd examples
$ python run_test.py

Further reading