sktime/pytorch-forecasting
Pytorch Forecasting
PyTorch Forecasting is a deep learning framework designed for building and training neural network architectures specifically for time series forecasting. It serves as a comprehensive toolkit for implementing autoregressive models, multi-horizon forecasting, and probabilistic prediction intervals using PyTorch tensors.
The library distinguishes itself through a probabilistic forecasting toolkit that generates prediction intervals and quantile forecasts using both parametric and non-parametric distributions. It further provides a neural network model optimizer for automated hyperparameter tuning and pruning to improve architecture efficiency.
The framework covers a broad surface of capabilities, including multivariate time series analysis and the integration of static and time-varying covariates. It includes a dedicated data pipeline for transforming tabular data into normalized tensors, as well as tools for interpretable model evaluation via variable dependency analysis and multi-horizon accuracy metrics.
The system supports distributed training across CPUs and multiple GPUs to accelerate model convergence.
Features
- Deep Learning Forecasting - Provides a comprehensive framework for applying deep neural networks to univariate and multivariate time series prediction.
- Autoregressive Models - Implements neural network architectures that generate long-term forecasts by iteratively feeding previous outputs back as inputs.
- Probabilistic - Enables probabilistic forecasting by mapping network outputs to probability distribution parameters instead of single point estimates.
- Probabilistic Loss Functions - Implements loss functions that map network outputs to probability distribution parameters for uncertainty quantification.
- Quantile Regression - Produces non-parametric probabilistic forecasts by predicting specific target distribution percentiles using pinball loss.
- Time Series Feature Engineering - Prepares raw tabular data for neural networks through automated scaling, normalization, and creation of training subsequences.
- Time Series Forecasting - A comprehensive deep learning framework for building and training PyTorch-based time series forecasting architectures.
- Dynamic Covariate Integration - Integrates static metadata and time-varying external variables into the model to improve prediction accuracy.
- Multivariate Forecasting - Predicts multiple interdependent time series variables simultaneously while incorporating external covariates.
- Probabilistic Forecasting - Predicts future demand with uncertainty intervals by generating probability distributions instead of point estimates.
- Time Series Tensor Pipelines - Converts tabular time series data into PyTorch tensors while automating scaling and feature encoding.
- Time Series Transformations - Transforms tabular time series data into normalized PyTorch tensors with configured encoder and decoder windows.
- Distributed Training - Supports spreading training processes across multiple CPUs or GPUs to accelerate model convergence.
- Covariate Integration - Incorporates static metadata and time-varying external variables into models to increase forecasting precision.
- Forecasting Interpretation Tools - Analyzes forecasting performance using multi-horizon metrics, attention maps, and variable dependency charts to understand model behavior.
- Hyperparameter Tuning - Optimizes model configuration settings using an automated search framework and pruning to improve training efficiency.
- Multi-Horizon Evaluation Metrics - Calculates performance metrics across various future time steps to measure accuracy over a prediction window.
- Multi-Target Learning - Simultaneously predicts multiple target variables using a combination of regression and classification.
- Hyperparameter Optimizers - Offers an automated hyperparameter tuning and pruning framework to optimize deep learning architectures.
- Prediction Visualization - Generates actual-versus-prediction plots and dependency charts to interpret model behavior and performance.
- Forecast Evaluation - Provides multi-horizon time series metrics to measure forecasting accuracy across various prediction windows.
- General Demand Forecasting - Provides architectures specifically designed to handle complex time series patterns for demand prediction.
- ML-Based Forecasting Models - Provides specialized machine learning architectures for real-world forecasting with built-in interpretation capabilities.
- Multivariate Quantile Prediction - Produces probabilistic forecasts across multiple variables and long horizons using hierarchical sampling.
- Neural Basis Expansion Analysis - Provides interpretable time series forecasting through basis-function decomposition, specifically supporting architectures like N-BEATS.
- Training Batch Generators - Samples overlapping encoder and decoder windows from time series to produce structured training batches.
- Variable Dependency Analysis - Calculates partial dependency for specific variables to analyze their influence on forecasting predictions.
- Interpretable - Generates explainable forecasts using basis function decomposition to clarify model behavior.
- Time Series Data Loading - Converts data tables into tensors while automating variable scaling, target normalization, and feature encoding.
- Time Series Data Normalization - Scales input data using customizable encoder methods and robust scaling via specific quantiles.
- Sliding Window Algorithms - Generates overlapping encoder and decoder windows from long time series to create training batches.
- Time Series Analysis - Deep learning-based time series forecasting using PyTorch.
- Time Series Analysis - PyTorch-based library for neural network forecasting.
