Need help with transformer-time-series-prediction?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

oliverguhr
366 Stars 96 Forks MIT License 9 Commits 2 Opened issues

Description

proof of concept for a transformer-based time series prediction model

Services available

!
?

Need anything else?

Contributors list

No Data

Transformer Time Series Prediction

This repository contains two Pytorch models for transformer-based time series prediction. Note that this is just a proof of concept and most likely not bug free nor particularly efficient.

  • transformer-singlestep.py contains a single-step prediction model
  • transformer-multistep.py contains a multi-step prediction model

This is the output of a single step prediction model that has been trained for 100 epochs on the daily minimum temperature dataset.

For any serious use case, head over to the flow-forecast package.

image

The model is also capable of multi-step predictions. In this case, the model predicted 200 steps based on 100 steps input. It was able to learn long term trend from the training data:

image

(input blue, prediction red)

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.