Nixtla

Neural 🧠 Forecast
Deep Learning for time series
State-of-the-art time series forecasting for PyTorch.
NeuralForecast is a Python library for time series forecasting with deep learning models. It includes benchmark datasets, data-loading utilities, evaluation functions, statistical tests, univariate model benchmarks and SOTA models implemented in PyTorch and PyTorchLightning.
⚡ Why?
Accuracy:
- Global model is fitted simultaneously for several time series.
- Shared information helps with highly parametrized and flexible models.
- Useful for items/skus that have little to no history available.
Efficiency:
- Automatic featurization processes.
- Fast computations (GPU or TPU).
📖 Documentation
Here is a link to the documentation.
🧬 Getting Started 
💻 Installation
PyPI
You can install the released version of NeuralForecast from the Python package index with:
pip install neuralforecast
(Installing inside a python virtualenvironment or a conda environment is recommended.)
Conda
Also you can install the released version of NeuralForecast from conda with:
conda install -c nixtla neuralforecast
(Installing inside a python virtualenvironment or a conda environment is recommended.)
Dev Mode
If you want to make some modifications to the code and see the effects in real time (without reinstalling), follow the steps below:git clone https://github.com/Nixtla/neuralforecast.git
cd neuralforecast
pip install -e .
Forecasting models
- Neural Hierarchical Interpolation for Time Series Forecasting (N-HiTS): A new model for long-horizon forecasting which incorporates novel hierarchical interpolation and multi-rate data sampling techniques to specialize blocks of its architecture to different frequency band of the time-series signal. It achieves SoTA performance on several benchmark datasets, outperforming current Transformer-based models by more than 25%.
- Exponential Smoothing Recurrent Neural Network (ES-RNN): A hybrid model that combines the expressivity of non linear models to capture the trends while it normalizes using a Holt-Winters inspired model for the levels and seasonals. This model is the winner of the M4 forecasting competition.
- Neural Basis Expansion Analysis (N-BEATS): A model from Element-AI (Yoshua Bengio’s lab) that has proven to achieve state-of-the-art performance on benchmark large scale forecasting datasets like Tourism, M3, and M4. The model is fast to train and has an interpretable configuration.
- Neural Basis Expansion Analysis with Exogenous Variables (N-BEATSx): The neural basis expansion with exogenous variables is an extension to the original N-BEATS that allows it to include time dependent covariates.
- Transformer-Based Models: Transformer-based framework for unsupervised representation learning of multivariate time series.
- Autoformer: Encoder-decoder model with decomposition capabilities and an approximation to attention based on Fourier transform.
- Informer: Transformer with MLP based multi-step prediction strategy, that approximates self-attention with sparsity.
- Transformer: Classical vanilla Transformer.
📃 License
This project is licensed under the GPLv3 License - see the LICENSE file for details.
🔨 How to contribute
See CONTRIBUTING.md.
Contributors ✨
Thanks goes to these wonderful people (emoji key):
fede 💻 🐛 📖 |
Greg DeVos 🤔 |
Cristian Challu 💻 |
mergenthaler 📖 💻 |
Kin 💻 🐛 🔣 |
José Morales 💻 |
Alejandro 💻 |
stefanialvs 🎨 |
Ikko Ashimine 🐛 |
vglaucus 🐛 |
Pietro Monticone 🐛 |
This project follows the all-contributors specification. Contributions of any kind welcome!
Release files for neuralforecast 0.0.7
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| neuralforecast-0.0.7.tar.gz | 109.2 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| neuralforecast-0.0.7-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 245.4 kB
Release files / neuralforecast-0.0.7.tar.gz
| Download URL | neuralforecast-0.0.7.tar.gz |
|---|---|
| Size | 109.2 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
31e27969195f03b9ccfb6cab39b339c8c1d52d4048ce727d3a6ffab1dceb2e94
|
|
BLAKE2b-256 checksum How to use checksums |
ef24bb9e845483ce103cb005cfb7004a80d16c97537425da99befecaa4c99f7f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.8.0 pkginfo/1.8.2 readme-renderer/32.0 requests/2.27.1 requests-toolbelt/0.9.1 urllib3/1.26.8 tqdm/4.62.3 importlib-metadata/4.10.1 keyring/23.5.0 rfc3986/2.0.0 colorama/0.4.4 CPython/3.7.12
|
Release files / neuralforecast-0.0.7-py3-none-any.whl
| Download URL | neuralforecast-0.0.7-py3-none-any.whl |
|---|---|
| Size | 136.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
bdb7f1b4a1148b7d7b4decd970759f94caac8ee543023331c72753e148a7461b
|
|
BLAKE2b-256 checksum How to use checksums |
f18655a7b9bbaac39d2698934eb02b72e8de4c48354e1b1a45c3a97e39ec1ee5
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.8.0 pkginfo/1.8.2 readme-renderer/32.0 requests/2.27.1 requests-toolbelt/0.9.1 urllib3/1.26.8 tqdm/4.62.3 importlib-metadata/4.10.1 keyring/23.5.0 rfc3986/2.0.0 colorama/0.4.4 CPython/3.7.12
|