temporal_fusion_transformer#

The temporal fusion transformer is a powerful predictive model for forecasting timeseries

Classes

TemporalFusionTransformer([hidden_size, ...])

Temporal Fusion Transformer for forecasting timeseries - use its from_dataset() method if possible.

Modules

pytorch_forecasting.models.temporal_fusion_transformer.sub_modules

Implementation of nn.Modules for temporal fusion transformer.

pytorch_forecasting.models.temporal_fusion_transformer.tuning

Hyperparameters can be efficiently tuned with optuna.