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.

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:

(input blue, prediction red)