Skip to content

Implementation of Transformer model (originally from Attention is All You Need) applied to Time Series.

License

Notifications You must be signed in to change notification settings

DanielAtKrypton/time_series_transformer

 
 

Repository files navigation

Transformers for Time Series

PyPI version travis codecov GitHub license Requirements Status

Based on timeseriestransformer.

Installation

Windows:

.\scripts\init-env.ps1

MacOS / Linux

./scripts/init-env.sh

Usage

from flights_time_series_dataset import FlightsDataset
from time_series_predictor import TimeSeriesPredictor
from time_series_transformer import Transformer

tsp = TimeSeriesPredictor(
    Transformer(),
    max_epochs=50,
    train_split=None,
)

tsp.fit(FlightsDataset())

Test

To test the package simply run the following command from project's root folder.

pytest -s

About

Implementation of Transformer model (originally from Attention is All You Need) applied to Time Series.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 96.3%
  • Shell 1.9%
  • PowerShell 1.8%