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.
Star history
How this analysis was created: This summary and feature list are AI-generated from collected project material and can contain mistakes. Stars, license and language are imported from GitHub. Inclusion does not mean that we have tested or audited this project. Check the source documentation for any feature you depend on. Learn more on our About page.
