Skip to main content

tft

Perform TFT forecast (Temporal Fusion Transformer): https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tft_model.html

Usage​

usage: tft [--lstm-layers LSTM_LAYERS] [--num-attention-heads NUM_ATTENTION_HEADS] [--full-attention] [--hidden-continuous-size HIDDEN_CONTINUOUS_SIZE]

Parameters​

NameDescriptionDefaultOptionalChoices
lstm_layersNumber of LSTM layers.1TrueNone
num_attention_headsNumber of attention heads.4TrueNone
full_attentionWhether to apply a multi-head attention query.FalseTrueNone
hidden_continuous_sizeDefault hidden size for processing continuous variables.8TrueNone

On this page