tuning

Hyperparameters can be efficiently tuned with optuna.

Functions

optimize_hyperparameters(train_dataloader, …)

Optimize Temporal Fusion Transformer hyperparameters.

Classes

MetricsCallback()

PyTorch Lightning metric callback.

VariableSelectionNetwork pytorch_forecasting.models.temporal_fusion_transformer.tuning.optimize_hyperparameters