youjp / Time-series-prediction

A collection of time series prediction methods: rnn, seq2seq, cnn, wavenet, transformer, unet, n-beats, gan, kalman-filter

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

RNN

intro

code

wavenet

intro

code

transformer

intro

code

U-Net

intro

code

n-beats

intro

code

GAN

intro

code

Usage

  1. Install the required library
pip install -r requirements.txt
  1. Download the data, if necessary
bash ./data/download_passenger.sh
  1. Train the model
    set custom_model_params if you want (refer to params in ./deepts/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

Contributor

About

A collection of time series prediction methods: rnn, seq2seq, cnn, wavenet, transformer, unet, n-beats, gan, kalman-filter

License:MIT License


Languages

Language:Python 99.7%Language:Shell 0.3%