# Time Series Made Easy in Python
![darts](https://github.com/unit8co/darts/raw/master/static/images/darts-logo-trim.png "darts")
---
[![PyPI version](https://badge.fury.io/py/u8darts.svg)](https://badge.fury.io/py/darts)
[![Conda Version](https://img.shields.io/conda/vn/conda-forge/u8darts-all.svg)](https://anaconda.org/conda-forge/u8darts-all)
![Supported versions](https://img.shields.io/badge/python-3.9+-blue.svg)
[![Docker Image Version (latest by date)](https://img.shields.io/docker/v/unit8/darts?label=docker&sort=date)](https://hub.docker.com/r/unit8/darts)
![GitHub Release Date](https://img.shields.io/github/release-date/unit8co/darts)
![GitHub Workflow Status](https://img.shields.io/github/actions/workflow/status/unit8co/darts/release.yml?branch=master)
[![Downloads](https://pepy.tech/badge/darts)](https://pepy.tech/project/darts)
[![Downloads](https://pepy.tech/badge/u8darts)](https://pepy.tech/project/u8darts)
[![codecov](https://codecov.io/gh/unit8co/darts/branch/master/graph/badge.svg?token=7F1TLUFHQW)](https://codecov.io/gh/unit8co/darts)
[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) [![Join the chat at https://gitter.im/u8darts/darts](https://badges.gitter.im/u8darts/darts.svg)](https://gitter.im/u8darts/darts?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge)
**Darts** is a Python library for user-friendly forecasting and anomaly detection
on time series. It contains a variety of models, from classics such as ARIMA to
deep neural networks. The forecasting models can all be used in the same way,
using `fit()` and `predict()` functions, similar to scikit-learn.
The library also makes it easy to backtest models,
combine the predictions of several models, and take external data into account.
Darts supports both univariate and multivariate time series and models.
The ML-based models can be trained on potentially large datasets containing multiple time
series, and some of the models offer a rich support for probabilistic forecasting.
Darts also offers extensive anomaly detection capabilities.
For instance, it is trivial to apply PyOD models on time series to obtain anomaly scores,
or to wrap any of Darts forecasting or filtering models to obtain fully
fledged anomaly detection models.
## Documentation
* [Quickstart](https://unit8co.github.io/darts/quickstart/00-quickstart.html)
* [User Guide](https://unit8co.github.io/darts/userguide.html)
* [API Reference](https://unit8co.github.io/darts/generated_api/darts.html)
* [Examples](https://unit8co.github.io/darts/examples.html)
##### High Level Introductions
* [Introductory Blog Post](https://medium.com/unit8-machine-learning-publication/darts-time-series-made-easy-in-python-5ac2947a8878)
* [Introduction video (25 minutes)](https://youtu.be/g6OXDnXEtFA)
##### Articles on Selected Topics
* [Training Models on Multiple Time Series](https://medium.com/unit8-machine-learning-publication/training-forecasting-models-on-multiple-time-series-with-darts-dc4be70b1844)
* [Using Past and Future Covariates](https://medium.com/unit8-machine-learning-publication/time-series-forecasting-using-past-and-future-external-data-with-darts-1f0539585993)
* [Temporal Convolutional Networks and Forecasting](https://medium.com/unit8-machine-learning-publication/temporal-convolutional-networks-and-forecasting-5ce1b6e97ce4)
* [Probabilistic Forecasting](https://medium.com/unit8-machine-learning-publication/probabilistic-forecasting-in-darts-e88fbe83344e)
* [Transfer Learning for Time Series Forecasting](https://medium.com/unit8-machine-learning-publication/transfer-learning-for-time-series-forecasting-87f39e375278)
* [Hierarchical Forecast Reconciliation](https://medium.com/unit8-machine-learning-publication/hierarchical-forecast-reconciliation-with-darts-8b4b058bb543)
## Quick Install
We recommend to first setup a clean Python environment for your project with Python 3.9+ using your favorite tool
([conda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html "conda-env"),
[venv](https://docs.python.org/3/library/venv.html), [virtualenv](https://virtualenv.pypa.io/en/latest/) with
or without [virtualenvwrapper](https://virtualenvwrapper.readthedocs.io/en/latest/)).
Once your environment is set up you can install darts using pip:
pip install darts
For more details you can refer to our
[installation instructions](https://github.com/unit8co/darts/blob/master/INSTALL.md).
## Example Usage
### Forecasting
Create a `TimeSeries` object from a Pandas DataFrame, and split it in train/validation series:
```python
import pandas as pd
from darts import TimeSeries
# Read a pandas DataFrame
df = pd.read_csv("AirPassengers.csv", delimiter=",")
# Create a TimeSeries, specifying the time and value columns
series = TimeSeries.from_dataframe(df, "Month", "#Passengers")
# Set aside the last 36 months as a validation series
train, val = series[:-36], series[-36:]
```
Fit an exponential smoothing model, and make a (probabilistic) prediction over the validation series' duration:
```python
from darts.models import ExponentialSmoothing
model = ExponentialSmoothing()
model.fit(train)
prediction = model.predict(len(val), num_samples=1000)
```
Plot the median, 5th and 95th percentiles:
```python
import matplotlib.pyplot as plt
series.plot()
prediction.plot(label="forecast", low_quantile=0.05, high_quantile=0.95)
plt.legend()
```
<div style="text-align:center;">
<img src="https://github.com/unit8co/darts/raw/master/static/images/example.png" alt="darts forecast example" />
</div>
### Anomaly Detection
Load a multivariate series, trim it, keep 2 components, split train and validation sets:
```python
from darts.datasets import ETTh2Dataset
series = ETTh2Dataset().load()[:10000][["MUFL", "LULL"]]
train, val = series.split_before(0.6)
```
Build a k-means anomaly scorer, train it on the train set
and use it on the validation set to get anomaly scores:
```python
from darts.ad import KMeansScorer
scorer = KMeansScorer(k=2, window=5)
scorer.fit(train)
anom_score = scorer.score(val)
```
Build a binary anomaly detector and train it over train scores,
then use it over validation scores to get binary anomaly classification:
```python
from darts.ad import QuantileDetector
detector = QuantileDetector(high_quantile=0.99)
detector.fit(scorer.score(train))
binary_anom = detector.detect(anom_score)
```
Plot (shifting and scaling some of the series
to make everything appear on the same figure):
```python
import matplotlib.pyplot as plt
series.plot()
(anom_score / 2. - 100).plot(label="computed anomaly score", c="orangered", lw=3)
(binary_anom * 45 - 150).plot(label="detected binary anomaly", lw=4)
```
<div style="text-align:center;">
<img src="https://github.com/unit8co/darts/raw/master/static/images/example_ad.png" alt="darts anomaly detection example" />
</div>
## Features
* **Forecasting Models:** A large collection of forecasting models; from statistical models (such as
ARIMA) to deep learning models (such as N-BEATS). See [table of models below](#forecasting-models).
* **Anomaly Detection** The `darts.ad` module contains a collection of anomaly scorers,
detectors and aggregators, which can all be combined to detect anomalies in time series.
It is easy to wrap any of Darts forecasting or filtering models to build
a fully fledged anomaly detection model that compares predictions with actuals.
The `PyODScorer` makes it trivial to use PyOD detectors on time series.
* **Multivariate Support:** `TimeSeries` can be multivariate - i.e., contain multiple time-varying
dimensions/columns instead of a single scalar value. Many models can consume and produce multivariate series.
* **Multiple Series Training (Global Models):** All machine learning based models (incl. all neural networks)
support being trained on multiple (potentially multivariate) series. This can scale to large datasets too.
* **Probabilistic Support:** `TimeSeries` objects can (optionally) represent stochastic
time series; this can for instance be used to get confidence intervals, and many models support different
flavours of probabilistic forecasting (such as estimating parametric distributions or quantiles).
Some anomaly detection scorers are also able to exploit these predictive distributions.
* **Conformal Prediction Support:** Our conformal prediction models allow to generate probabilistic forecasts with
calibrated quantile intervals for any pre-trained global forecasting model.
* **Past and Future Covariates Support:** Many models in Darts support past-observed and/or future-known
covariate (external data) time series as inputs for producing forecasts.
* **Static Covariates Support:** In addition to time-dependent data, `TimeSeries` can also contain
static data for each dimension, which can be exploited by some models.
* **Hierarchical Reconciliation:** Darts offers transformers to perform reconciliation.
These can make the forecasts add up in a way that respects the underlying hierarchy.
* **Regression Models:** It is possible to plug-in any scikit-learn compatible model
to obtain forecasts as functions of lagged values of the target series and covariates.
* **Training with Sample Weights:** All global models support being trained with sample weights. They can be
applied to each observation, forecasted time step and target column.
* **Forecast Start Shifting:** All global models support training and prediction on a shifted output window.
This is useful for example for Day-Ahead Market forecasts, or when the covariates (or target series) are reported
with a delay.
* **Explainability:** Darts has the ability to *explain* some forecasting models using Shap values.
* **Data Processing:** Tools to easily apply (and revert) common transformations on
time series data (scaling, filling missing values, differencing, boxcox, ...)
* **Metrics:** A variety of metrics for evaluating time series' goodness of fit;
from R2-scores to Mean Absolute Scaled Error.
* **Backtesting:** Utilities for simulating historical forecasts, using moving time windows.
* **PyTorch Lightning Support:** All deep learning models are implemented using PyTorch Lightning,
supporting among other things custom callbacks, GPUs/TPUs training and custom trainers.
* **Filtering Models:** Darts offers three filtering models: `KalmanFilter`, `GaussianProcessFilter`,
and `MovingAverageFilter`, which allow to filter time series, and in some cases obtain probabilistic
inferences of the underlying states/values.
* **Datasets** The `darts.datasets` submodule contains some popular time series datasets for rapid
and reproducible experimentation.
## Forecasting Models
Here's a breakdown of the forecasting models currently implemented in Darts. We are constantly working
on bringing more models and features.
| Model | Sources | Target Series Support:<br/><br/>Univariate/<br/>Multivariate | Covariates Support:<br/><br/>Past-observed/<br/>Future-known/<br/>Static | Probabilistic Forecasting:<br/><br/>Sampled/<br/>Distribution Parameters | Training & Forecasting on Multiple Series |
|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------|--------------------------------------------------------------------------|--------------------------------------------------------------------------|-------------------------------------------|
| **Baseline Models**<br/>([LocalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#local-forecasting-models-lfms)) | | | | | |
| [NaiveMean](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveMean) | | β
β
| π΄ π΄ π΄ | π΄ π΄ | π΄ |
| [NaiveSeasonal](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveSeasonal) | | β
β
| π΄ π΄ π΄ | π΄ π΄ | π΄ |
| [NaiveDrift](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveDrift) | | β
β
| π΄ π΄ π΄ | π΄ π΄ | π΄ |
| [NaiveMovingAverage](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveMovingAverage) | | β
β
| π΄ π΄ π΄ | π΄ π΄ | π΄ |
| **Statistical / Classic Models**<br/>([LocalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#local-forecasting-models-lfms)) | | | | | |
| [ARIMA](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.arima.html#darts.models.forecasting.arima.ARIMA) | | β
π΄ | π΄ β
π΄ | β
π΄ | π΄ |
| [VARIMA](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.varima.html#darts.models.forecasting.varima.VARIMA) | | π΄ β
| π΄ β
π΄ | β
π΄ | π΄ |
| [AutoARIMA](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.auto_arima.html#darts.models.forecasting.auto_arima.AutoARIMA) | | β
π΄ | π΄ β
π΄ | π΄ π΄ | π΄ |
| [StatsForecastAutoArima](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_arima.html#darts.models.forecasting.sf_auto_arima.StatsForecastAutoARIMA) (faster AutoARIMA) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | β
π΄ | π΄ β
π΄ | β
π΄ | π΄ |
| [ExponentialSmoothing](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.exponential_smoothing.html#darts.models.forecasting.exponential_smoothing.ExponentialSmoothing) | | β
π΄ | π΄ π΄ π΄ | β
π΄ | π΄ |
| [StatsforecastAutoETS](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_ets.html#darts.models.forecasting.sf_auto_ets.StatsForecastAutoETS) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | β
π΄ | π΄ β
π΄ | β
π΄ | π΄ |
| [StatsforecastAutoCES](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_ces.html#darts.models.forecasting.sf_auto_ces.StatsForecastAutoCES) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | β
π΄ | π΄ π΄ π΄ | π΄ π΄ | π΄ |
| [BATS](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tbats_model.html#darts.models.forecasting.tbats_model.BATS) and [TBATS](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tbats_model.html#darts.models.forecasting.tbats_model.TBATS) | [TBATS paper](https://robjhyndman.com/papers/ComplexSeasonality.pdf) | β
π΄ | π΄ π΄ π΄ | β
π΄ | π΄ |
| [Theta](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.theta.html#darts.models.forecasting.theta.Theta) and [FourTheta](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.theta.html#darts.models.forecasting.theta.FourTheta) | [Theta](https://robjhyndman.com/papers/Theta.pdf) & [4 Theta](https://github.com/Mcompetitions/M4-methods/blob/master/4Theta%20method.R) | β
π΄ | π΄ π΄ π΄ | π΄ π΄ | π΄ |
| [StatsForecastAutoTheta](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_theta.html#darts.models.forecasting.sf_auto_theta.StatsForecastAutoTheta) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | β
π΄ | π΄ π΄ π΄ | β
π΄ | π΄ |
| [Prophet](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.prophet_model.html#darts.models.forecasting.prophet_model.Prophet) | [Prophet repo](https://github.com/facebook/prophet) | β
π΄ | π΄ β
π΄ | β
π΄ | π΄ |
| [FFT](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.fft.html#darts.models.forecasting.fft.FFT) (Fast Fourier Transform) | | β
π΄ | π΄ π΄ π΄ | π΄ π΄ | π΄ |
| [KalmanForecaster](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.kalman_forecaster.html#darts.models.forecasting.kalman_forecaster.KalmanForecaster) using the Kalman filter and N4SID for system identification | [N4SID paper](https://people.duke.edu/~hpgavin/SystemID/References/VanOverschee-Automatica-1994.pdf) | β
β
| π΄ β
π΄ | β
π΄ | π΄ |
| [Croston](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.croston.html#darts.models.forecasting.croston.Croston) method | | β
π΄ | π΄ π΄ π΄ | π΄ π΄ | π΄ |
| **Global Baseline Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)) | | | | | |
| [GlobalNaiveAggregate](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.global_baseline_models.html#darts.models.forecasting.global_baseline_models.GlobalNaiveAggregate) | | β
β
| π΄ π΄ π΄ | π΄ π΄ | β
|
| [GlobalNaiveDrift](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.global_baseline_models.html#darts.models.forecasting.global_baseline_models.GlobalNaiveDrift) | | β
β
| π΄ π΄ π΄ | π΄ π΄ | β
|
| [GlobalNaiveSeasonal](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.global_baseline_models.html#darts.models.forecasting.global_baseline_models.GlobalNaiveSeasonal) | | β
β
| π΄ π΄ π΄ | π΄ π΄ | β
|
| **Regression Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)) | | | | | |
| [RegressionModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.regression_model.html#darts.models.forecasting.regression_model.RegressionModel): generic wrapper around any sklearn regression model | | β
β
| β
β
β
| π΄ π΄ | β
|
| [LinearRegressionModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.linear_regression_model.html#darts.models.forecasting.linear_regression_model.LinearRegressionModel) | | β
β
| β
β
β
| β
β
| β
|
| [RandomForest](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.random_forest.html#darts.models.forecasting.random_forest.RandomForest) | | β
β
| β
β
β
| π΄ π΄ | β
|
| [LightGBMModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.lgbm.html#darts.models.forecasting.lgbm.LightGBMModel) | | β
β
| β
β
β
| β
β
| β
|
| [XGBModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.xgboost.html#darts.models.forecasting.xgboost.XGBModel) | | β
β
| β
β
β
| β
β
| β
|
| [CatBoostModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.catboost_model.html#darts.models.forecasting.catboost_model.CatBoostModel) | | β
β
| β
β
β
| β
β
| β
|
| **PyTorch (Lightning)-based Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)) | | | | | |
| [RNNModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.rnn_model.html#darts.models.forecasting.rnn_model.RNNModel) (incl. LSTM and GRU); equivalent to DeepAR in its probabilistic version | [DeepAR paper](https://arxiv.org/abs/1704.04110) | β
β
| π΄ β
π΄ | β
β
| β
|
| [BlockRNNModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.block_rnn_model.html#darts.models.forecasting.block_rnn_model.BlockRNNModel) (incl. LSTM and GRU) | | β
β
| β
π΄ π΄ | β
β
| β
|
| [NBEATSModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.nbeats.html#darts.models.forecasting.nbeats.NBEATSModel) | [N-BEATS paper](https://arxiv.org/abs/1905.10437) | β
β
| β
π΄ π΄ | β
β
| β
|
| [NHiTSModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.nhits.html#darts.models.forecasting.nhits.NHiTSModel) | [N-HiTS paper](https://arxiv.org/abs/2201.12886) | β
β
| β
π΄ π΄ | β
β
| β
|
| [TCNModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tcn_model.html#darts.models.forecasting.tcn_model.TCNModel) | [TCN paper](https://arxiv.org/abs/1803.01271), [DeepTCN paper](https://arxiv.org/abs/1906.04397), [blog post](https://medium.com/unit8-machine-learning-publication/temporal-convolutional-networks-and-forecasting-5ce1b6e97ce4) | β
β
| β
π΄ π΄ | β
β
| β
|
| [TransformerModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.transformer_model.html#darts.models.forecasting.transformer_model.TransformerModel) | | β
β
| β
π΄ π΄ | β
β
| β
|
| [TFTModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tft_model.html#darts.models.forecasting.tft_model.TFTModel) (Temporal Fusion Transformer) | [TFT paper](https://arxiv.org/pdf/1912.09363.pdf), [PyTorch Forecasting](https://pytorch-forecasting.readthedocs.io/en/latest/models.html) | β
β
| β
β
β
| β
β
| β
|
| [DLinearModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.dlinear.html#darts.models.forecasting.dlinear.DLinearModel) | [DLinear paper](https://arxiv.org/pdf/2205.13504.pdf) | β
β
| β
β
β
| β
β
| β
|
| [NLinearModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.nlinear.html#darts.models.forecasting.nlinear.NLinearModel) | [NLinear paper](https://arxiv.org/pdf/2205.13504.pdf) | β
β
| β
β
β
| β
β
| β
|
| [TiDEModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tide_model.html#darts.models.forecasting.tide_model.TiDEModel) | [TiDE paper](https://arxiv.org/pdf/2304.08424.pdf) | β
β
| β
β
β
| β
β
| β
|
| [TSMixerModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tsmixer_model.html#darts.models.forecasting.tsmixer_model.TSMixerModel) | [TSMixer paper](https://arxiv.org/pdf/2303.06053.pdf), [PyTorch Implementation](https://github.com/ditschuk/pytorch-tsmixer) | β
β
| β
β
β
| β
β
| β
|
| **Ensemble Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)): Model support is dependent on ensembled forecasting models and the ensemble model itself | | | | | |
| [NaiveEnsembleModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveEnsembleModel) | | β
β
| β
β
β
| β
β
| β
|
| [RegressionEnsembleModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.regression_ensemble_model.html#darts.models.forecasting.regression_ensemble_model.RegressionEnsembleModel) | | β
β
| β
β
β
| β
β
| β
|
| **Conformal Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)): Model support is dependent on the forecasting model used | | | | | |
| [ConformalNaiveModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.conformal_models.html#darts.models.forecasting.conformal_models.ConformalNaiveModel) | [Conformalized Prediction](https://arxiv.org/pdf/1905.03222) | β
β
| β
β
β
| β
β
| β
|
| [ConformalQRModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.conformal_models.html#darts.models.forecasting.conformal_models.ConformalQRModel) | [Conformalized Quantile Regression](https://arxiv.org/pdf/1905.03222) | β
β
| β
β
β
| β
β
| β
|
## Community & Contact
Anyone is welcome to join our [Gitter room](https://gitter.im/u8darts/darts) to ask questions, make proposals,
discuss use-cases, and more. If you spot a bug or have suggestions, GitHub issues are also welcome.
If what you want to tell us is not suitable for Gitter or Github,
feel free to send us an email at <a href="mailto:darts@unit8.co">darts@unit8.co</a> for
darts related matters or <a href="mailto:info@unit8.co">info@unit8.co</a> for any other
inquiries.
## Contribute
The development is ongoing, and we welcome suggestions, pull requests and issues on GitHub.
All contributors will be acknowledged on the
[change log page](https://github.com/unit8co/darts/blob/master/CHANGELOG.md).
Before working on a contribution (a new feature or a fix),
[check our contribution guidelines](https://github.com/unit8co/darts/blob/master/CONTRIBUTING.md).
## Citation
If you are using Darts in your scientific work, we would appreciate citations to the following JMLR paper.
[Darts: User-Friendly Modern Machine Learning for Time Series](https://www.jmlr.org/papers/v23/21-1177.html)
Bibtex entry:
```
@article{JMLR:v23:21-1177,
author = {Julien Herzen and Francesco LΓΒ€ssig and Samuele Giuliano Piazzetta and Thomas Neuer and LΓΒ©o Tafti and Guillaume Raille and Tomas Van Pottelbergh and Marek Pasieka and Andrzej Skrodzki and Nicolas Huguenin and Maxime Dumonal and Jan KoΓ
βΊcisz and Dennis Bader and FrΓΒ©dΓΒ©rick Gusset and Mounir Benheddi and Camila Williamson and Michal Kosinski and Matej Petrik and GaΓΒ«l Grosch},
title = {Darts: User-Friendly Modern Machine Learning for Time Series},
journal = {Journal of Machine Learning Research},
year = {2022},
volume = {23},
number = {124},
pages = {1-6},
url = {http://jmlr.org/papers/v23/21-1177.html}
}
```
Raw data
{
"_id": null,
"home_page": "https://unit8co.github.io/darts/",
"name": "darts",
"maintainer": "Unit8 SA",
"docs_url": null,
"requires_python": ">=3.9",
"maintainer_email": "darts@unit8.co",
"keywords": "time series forecasting",
"author": null,
"author_email": null,
"download_url": "https://files.pythonhosted.org/packages/a8/f2/7bde3ddcac7ed47c545ec725bfa28e654cd52ac7c683edb7dbe97fe603a0/darts-0.32.0.tar.gz",
"platform": null,
"description": "# Time Series Made Easy in Python\n\n![darts](https://github.com/unit8co/darts/raw/master/static/images/darts-logo-trim.png \"darts\")\n\n---\n[![PyPI version](https://badge.fury.io/py/u8darts.svg)](https://badge.fury.io/py/darts)\n[![Conda Version](https://img.shields.io/conda/vn/conda-forge/u8darts-all.svg)](https://anaconda.org/conda-forge/u8darts-all)\n![Supported versions](https://img.shields.io/badge/python-3.9+-blue.svg)\n[![Docker Image Version (latest by date)](https://img.shields.io/docker/v/unit8/darts?label=docker&sort=date)](https://hub.docker.com/r/unit8/darts)\n![GitHub Release Date](https://img.shields.io/github/release-date/unit8co/darts)\n![GitHub Workflow Status](https://img.shields.io/github/actions/workflow/status/unit8co/darts/release.yml?branch=master)\n[![Downloads](https://pepy.tech/badge/darts)](https://pepy.tech/project/darts)\n[![Downloads](https://pepy.tech/badge/u8darts)](https://pepy.tech/project/u8darts)\n[![codecov](https://codecov.io/gh/unit8co/darts/branch/master/graph/badge.svg?token=7F1TLUFHQW)](https://codecov.io/gh/unit8co/darts)\n[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black) [![Join the chat at https://gitter.im/u8darts/darts](https://badges.gitter.im/u8darts/darts.svg)](https://gitter.im/u8darts/darts?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge)\n\n**Darts** is a Python library for user-friendly forecasting and anomaly detection\non time series. It contains a variety of models, from classics such as ARIMA to\ndeep neural networks. The forecasting models can all be used in the same way,\nusing `fit()` and `predict()` functions, similar to scikit-learn.\nThe library also makes it easy to backtest models,\ncombine the predictions of several models, and take external data into account.\nDarts supports both univariate and multivariate time series and models.\nThe ML-based models can be trained on potentially large datasets containing multiple time\nseries, and some of the models offer a rich support for probabilistic forecasting.\n\nDarts also offers extensive anomaly detection capabilities.\nFor instance, it is trivial to apply PyOD models on time series to obtain anomaly scores,\nor to wrap any of Darts forecasting or filtering models to obtain fully\nfledged anomaly detection models.\n\n\n## Documentation\n* [Quickstart](https://unit8co.github.io/darts/quickstart/00-quickstart.html)\n* [User Guide](https://unit8co.github.io/darts/userguide.html)\n* [API Reference](https://unit8co.github.io/darts/generated_api/darts.html)\n* [Examples](https://unit8co.github.io/darts/examples.html)\n\n##### High Level Introductions\n* [Introductory Blog Post](https://medium.com/unit8-machine-learning-publication/darts-time-series-made-easy-in-python-5ac2947a8878)\n* [Introduction video (25 minutes)](https://youtu.be/g6OXDnXEtFA)\n\n##### Articles on Selected Topics\n* [Training Models on Multiple Time Series](https://medium.com/unit8-machine-learning-publication/training-forecasting-models-on-multiple-time-series-with-darts-dc4be70b1844)\n* [Using Past and Future Covariates](https://medium.com/unit8-machine-learning-publication/time-series-forecasting-using-past-and-future-external-data-with-darts-1f0539585993)\n* [Temporal Convolutional Networks and Forecasting](https://medium.com/unit8-machine-learning-publication/temporal-convolutional-networks-and-forecasting-5ce1b6e97ce4)\n* [Probabilistic Forecasting](https://medium.com/unit8-machine-learning-publication/probabilistic-forecasting-in-darts-e88fbe83344e)\n* [Transfer Learning for Time Series Forecasting](https://medium.com/unit8-machine-learning-publication/transfer-learning-for-time-series-forecasting-87f39e375278)\n* [Hierarchical Forecast Reconciliation](https://medium.com/unit8-machine-learning-publication/hierarchical-forecast-reconciliation-with-darts-8b4b058bb543)\n\n## Quick Install\n\nWe recommend to first setup a clean Python environment for your project with Python 3.9+ using your favorite tool\n([conda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html \"conda-env\"),\n[venv](https://docs.python.org/3/library/venv.html), [virtualenv](https://virtualenv.pypa.io/en/latest/) with\nor without [virtualenvwrapper](https://virtualenvwrapper.readthedocs.io/en/latest/)).\n\nOnce your environment is set up you can install darts using pip:\n\n pip install darts\n\nFor more details you can refer to our\n[installation instructions](https://github.com/unit8co/darts/blob/master/INSTALL.md).\n\n## Example Usage\n\n### Forecasting\n\nCreate a `TimeSeries` object from a Pandas DataFrame, and split it in train/validation series:\n\n```python\nimport pandas as pd\nfrom darts import TimeSeries\n\n# Read a pandas DataFrame\ndf = pd.read_csv(\"AirPassengers.csv\", delimiter=\",\")\n\n# Create a TimeSeries, specifying the time and value columns\nseries = TimeSeries.from_dataframe(df, \"Month\", \"#Passengers\")\n\n# Set aside the last 36 months as a validation series\ntrain, val = series[:-36], series[-36:]\n```\n\nFit an exponential smoothing model, and make a (probabilistic) prediction over the validation series' duration:\n```python\nfrom darts.models import ExponentialSmoothing\n\nmodel = ExponentialSmoothing()\nmodel.fit(train)\nprediction = model.predict(len(val), num_samples=1000)\n```\n\nPlot the median, 5th and 95th percentiles:\n```python\nimport matplotlib.pyplot as plt\n\nseries.plot()\nprediction.plot(label=\"forecast\", low_quantile=0.05, high_quantile=0.95)\nplt.legend()\n```\n\n<div style=\"text-align:center;\">\n<img src=\"https://github.com/unit8co/darts/raw/master/static/images/example.png\" alt=\"darts forecast example\" />\n</div>\n\n### Anomaly Detection\n\nLoad a multivariate series, trim it, keep 2 components, split train and validation sets:\n\n```python\nfrom darts.datasets import ETTh2Dataset\n\nseries = ETTh2Dataset().load()[:10000][[\"MUFL\", \"LULL\"]]\ntrain, val = series.split_before(0.6)\n```\n\nBuild a k-means anomaly scorer, train it on the train set\nand use it on the validation set to get anomaly scores:\n\n```python\nfrom darts.ad import KMeansScorer\n\nscorer = KMeansScorer(k=2, window=5)\nscorer.fit(train)\nanom_score = scorer.score(val)\n```\n\nBuild a binary anomaly detector and train it over train scores,\nthen use it over validation scores to get binary anomaly classification:\n\n```python\nfrom darts.ad import QuantileDetector\n\ndetector = QuantileDetector(high_quantile=0.99)\ndetector.fit(scorer.score(train))\nbinary_anom = detector.detect(anom_score)\n```\n\nPlot (shifting and scaling some of the series\nto make everything appear on the same figure):\n\n```python\nimport matplotlib.pyplot as plt\n\nseries.plot()\n(anom_score / 2. - 100).plot(label=\"computed anomaly score\", c=\"orangered\", lw=3)\n(binary_anom * 45 - 150).plot(label=\"detected binary anomaly\", lw=4)\n```\n\n<div style=\"text-align:center;\">\n<img src=\"https://github.com/unit8co/darts/raw/master/static/images/example_ad.png\" alt=\"darts anomaly detection example\" />\n</div>\n\n\n## Features\n* **Forecasting Models:** A large collection of forecasting models; from statistical models (such as\n ARIMA) to deep learning models (such as N-BEATS). See [table of models below](#forecasting-models).\n\n* **Anomaly Detection** The `darts.ad` module contains a collection of anomaly scorers,\n detectors and aggregators, which can all be combined to detect anomalies in time series.\n It is easy to wrap any of Darts forecasting or filtering models to build\n a fully fledged anomaly detection model that compares predictions with actuals.\n The `PyODScorer` makes it trivial to use PyOD detectors on time series.\n\n* **Multivariate Support:** `TimeSeries` can be multivariate - i.e., contain multiple time-varying\n dimensions/columns instead of a single scalar value. Many models can consume and produce multivariate series.\n\n* **Multiple Series Training (Global Models):** All machine learning based models (incl. all neural networks)\n support being trained on multiple (potentially multivariate) series. This can scale to large datasets too.\n\n* **Probabilistic Support:** `TimeSeries` objects can (optionally) represent stochastic\n time series; this can for instance be used to get confidence intervals, and many models support different\n flavours of probabilistic forecasting (such as estimating parametric distributions or quantiles).\n Some anomaly detection scorers are also able to exploit these predictive distributions.\n\n* **Conformal Prediction Support:** Our conformal prediction models allow to generate probabilistic forecasts with\n calibrated quantile intervals for any pre-trained global forecasting model.\n\n* **Past and Future Covariates Support:** Many models in Darts support past-observed and/or future-known\n covariate (external data) time series as inputs for producing forecasts.\n\n* **Static Covariates Support:** In addition to time-dependent data, `TimeSeries` can also contain\n static data for each dimension, which can be exploited by some models.\n\n* **Hierarchical Reconciliation:** Darts offers transformers to perform reconciliation.\n These can make the forecasts add up in a way that respects the underlying hierarchy.\n\n* **Regression Models:** It is possible to plug-in any scikit-learn compatible model\n to obtain forecasts as functions of lagged values of the target series and covariates.\n\n* **Training with Sample Weights:** All global models support being trained with sample weights. They can be\n applied to each observation, forecasted time step and target column.\n\n* **Forecast Start Shifting:** All global models support training and prediction on a shifted output window.\n This is useful for example for Day-Ahead Market forecasts, or when the covariates (or target series) are reported\n with a delay.\n\n* **Explainability:** Darts has the ability to *explain* some forecasting models using Shap values.\n\n* **Data Processing:** Tools to easily apply (and revert) common transformations on\n time series data (scaling, filling missing values, differencing, boxcox, ...)\n\n* **Metrics:** A variety of metrics for evaluating time series' goodness of fit;\n from R2-scores to Mean Absolute Scaled Error.\n\n* **Backtesting:** Utilities for simulating historical forecasts, using moving time windows.\n\n* **PyTorch Lightning Support:** All deep learning models are implemented using PyTorch Lightning,\n supporting among other things custom callbacks, GPUs/TPUs training and custom trainers.\n\n* **Filtering Models:** Darts offers three filtering models: `KalmanFilter`, `GaussianProcessFilter`,\n and `MovingAverageFilter`, which allow to filter time series, and in some cases obtain probabilistic\n inferences of the underlying states/values.\n\n* **Datasets** The `darts.datasets` submodule contains some popular time series datasets for rapid\n and reproducible experimentation.\n\n## Forecasting Models\nHere's a breakdown of the forecasting models currently implemented in Darts. We are constantly working\non bringing more models and features.\n\n\n| Model | Sources | Target Series Support:<br/><br/>Univariate/<br/>Multivariate | Covariates Support:<br/><br/>Past-observed/<br/>Future-known/<br/>Static | Probabilistic Forecasting:<br/><br/>Sampled/<br/>Distribution Parameters | Training & Forecasting on Multiple Series |\n|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------|--------------------------------------------------------------------------|--------------------------------------------------------------------------|-------------------------------------------|\n| **Baseline Models**<br/>([LocalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#local-forecasting-models-lfms)) | | | | | |\n| [NaiveMean](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveMean) | | \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| [NaiveSeasonal](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveSeasonal) | | \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| [NaiveDrift](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveDrift) | | \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| [NaiveMovingAverage](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveMovingAverage) | | \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| **Statistical / Classic Models**<br/>([LocalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#local-forecasting-models-lfms)) | | | | | |\n| [ARIMA](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.arima.html#darts.models.forecasting.arima.ARIMA) | | \u2705 \ud83d\udd34 | \ud83d\udd34 \u2705 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [VARIMA](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.varima.html#darts.models.forecasting.varima.VARIMA) | | \ud83d\udd34 \u2705 | \ud83d\udd34 \u2705 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [AutoARIMA](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.auto_arima.html#darts.models.forecasting.auto_arima.AutoARIMA) | | \u2705 \ud83d\udd34 | \ud83d\udd34 \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| [StatsForecastAutoArima](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_arima.html#darts.models.forecasting.sf_auto_arima.StatsForecastAutoARIMA) (faster AutoARIMA) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | \u2705 \ud83d\udd34 | \ud83d\udd34 \u2705 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [ExponentialSmoothing](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.exponential_smoothing.html#darts.models.forecasting.exponential_smoothing.ExponentialSmoothing) | | \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [StatsforecastAutoETS](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_ets.html#darts.models.forecasting.sf_auto_ets.StatsForecastAutoETS) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | \u2705 \ud83d\udd34 | \ud83d\udd34 \u2705 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [StatsforecastAutoCES](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_ces.html#darts.models.forecasting.sf_auto_ces.StatsForecastAutoCES) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| [BATS](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tbats_model.html#darts.models.forecasting.tbats_model.BATS) and [TBATS](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tbats_model.html#darts.models.forecasting.tbats_model.TBATS) | [TBATS paper](https://robjhyndman.com/papers/ComplexSeasonality.pdf) | \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [Theta](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.theta.html#darts.models.forecasting.theta.Theta) and [FourTheta](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.theta.html#darts.models.forecasting.theta.FourTheta) | [Theta](https://robjhyndman.com/papers/Theta.pdf) & [4 Theta](https://github.com/Mcompetitions/M4-methods/blob/master/4Theta%20method.R) | \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| [StatsForecastAutoTheta](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.sf_auto_theta.html#darts.models.forecasting.sf_auto_theta.StatsForecastAutoTheta) | [Nixtla's statsforecast](https://github.com/Nixtla/statsforecast) | \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [Prophet](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.prophet_model.html#darts.models.forecasting.prophet_model.Prophet) | [Prophet repo](https://github.com/facebook/prophet) | \u2705 \ud83d\udd34 | \ud83d\udd34 \u2705 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [FFT](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.fft.html#darts.models.forecasting.fft.FFT) (Fast Fourier Transform) | | \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| [KalmanForecaster](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.kalman_forecaster.html#darts.models.forecasting.kalman_forecaster.KalmanForecaster) using the Kalman filter and N4SID for system identification | [N4SID paper](https://people.duke.edu/~hpgavin/SystemID/References/VanOverschee-Automatica-1994.pdf) | \u2705 \u2705 | \ud83d\udd34 \u2705 \ud83d\udd34 | \u2705 \ud83d\udd34 | \ud83d\udd34 |\n| [Croston](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.croston.html#darts.models.forecasting.croston.Croston) method | | \u2705 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 |\n| **Global Baseline Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)) | | | | | |\n| [GlobalNaiveAggregate](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.global_baseline_models.html#darts.models.forecasting.global_baseline_models.GlobalNaiveAggregate) | | \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \u2705 |\n| [GlobalNaiveDrift](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.global_baseline_models.html#darts.models.forecasting.global_baseline_models.GlobalNaiveDrift) | | \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \u2705 |\n| [GlobalNaiveSeasonal](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.global_baseline_models.html#darts.models.forecasting.global_baseline_models.GlobalNaiveSeasonal) | | \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 \ud83d\udd34 | \ud83d\udd34 \ud83d\udd34 | \u2705 |\n| **Regression Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)) | | | | | |\n| [RegressionModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.regression_model.html#darts.models.forecasting.regression_model.RegressionModel): generic wrapper around any sklearn regression model | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 | \u2705 |\n| [LinearRegressionModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.linear_regression_model.html#darts.models.forecasting.linear_regression_model.LinearRegressionModel) | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [RandomForest](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.random_forest.html#darts.models.forecasting.random_forest.RandomForest) | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \ud83d\udd34 \ud83d\udd34 | \u2705 |\n| [LightGBMModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.lgbm.html#darts.models.forecasting.lgbm.LightGBMModel) | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [XGBModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.xgboost.html#darts.models.forecasting.xgboost.XGBModel) | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [CatBoostModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.catboost_model.html#darts.models.forecasting.catboost_model.CatBoostModel) | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| **PyTorch (Lightning)-based Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)) | | | | | |\n| [RNNModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.rnn_model.html#darts.models.forecasting.rnn_model.RNNModel) (incl. LSTM and GRU); equivalent to DeepAR in its probabilistic version | [DeepAR paper](https://arxiv.org/abs/1704.04110) | \u2705 \u2705 | \ud83d\udd34 \u2705 \ud83d\udd34 | \u2705 \u2705 | \u2705 |\n| [BlockRNNModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.block_rnn_model.html#darts.models.forecasting.block_rnn_model.BlockRNNModel) (incl. LSTM and GRU) | | \u2705 \u2705 | \u2705 \ud83d\udd34 \ud83d\udd34 | \u2705 \u2705 | \u2705 |\n| [NBEATSModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.nbeats.html#darts.models.forecasting.nbeats.NBEATSModel) | [N-BEATS paper](https://arxiv.org/abs/1905.10437) | \u2705 \u2705 | \u2705 \ud83d\udd34 \ud83d\udd34 | \u2705 \u2705 | \u2705 |\n| [NHiTSModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.nhits.html#darts.models.forecasting.nhits.NHiTSModel) | [N-HiTS paper](https://arxiv.org/abs/2201.12886) | \u2705 \u2705 | \u2705 \ud83d\udd34 \ud83d\udd34 | \u2705 \u2705 | \u2705 |\n| [TCNModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tcn_model.html#darts.models.forecasting.tcn_model.TCNModel) | [TCN paper](https://arxiv.org/abs/1803.01271), [DeepTCN paper](https://arxiv.org/abs/1906.04397), [blog post](https://medium.com/unit8-machine-learning-publication/temporal-convolutional-networks-and-forecasting-5ce1b6e97ce4) | \u2705 \u2705 | \u2705 \ud83d\udd34 \ud83d\udd34 | \u2705 \u2705 | \u2705 |\n| [TransformerModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.transformer_model.html#darts.models.forecasting.transformer_model.TransformerModel) | | \u2705 \u2705 | \u2705 \ud83d\udd34 \ud83d\udd34 | \u2705 \u2705 | \u2705 |\n| [TFTModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tft_model.html#darts.models.forecasting.tft_model.TFTModel) (Temporal Fusion Transformer) | [TFT paper](https://arxiv.org/pdf/1912.09363.pdf), [PyTorch Forecasting](https://pytorch-forecasting.readthedocs.io/en/latest/models.html) | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [DLinearModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.dlinear.html#darts.models.forecasting.dlinear.DLinearModel) | [DLinear paper](https://arxiv.org/pdf/2205.13504.pdf) | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [NLinearModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.nlinear.html#darts.models.forecasting.nlinear.NLinearModel) | [NLinear paper](https://arxiv.org/pdf/2205.13504.pdf) | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [TiDEModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tide_model.html#darts.models.forecasting.tide_model.TiDEModel) | [TiDE paper](https://arxiv.org/pdf/2304.08424.pdf) | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [TSMixerModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.tsmixer_model.html#darts.models.forecasting.tsmixer_model.TSMixerModel) | [TSMixer paper](https://arxiv.org/pdf/2303.06053.pdf), [PyTorch Implementation](https://github.com/ditschuk/pytorch-tsmixer) | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| **Ensemble Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)): Model support is dependent on ensembled forecasting models and the ensemble model itself | | | | | |\n| [NaiveEnsembleModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.baselines.html#darts.models.forecasting.baselines.NaiveEnsembleModel) | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [RegressionEnsembleModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.regression_ensemble_model.html#darts.models.forecasting.regression_ensemble_model.RegressionEnsembleModel) | | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| **Conformal Models**<br/>([GlobalForecastingModel](https://unit8co.github.io/darts/userguide/covariates.html#global-forecasting-models-gfms)): Model support is dependent on the forecasting model used | | | | | |\n| [ConformalNaiveModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.conformal_models.html#darts.models.forecasting.conformal_models.ConformalNaiveModel) | [Conformalized Prediction](https://arxiv.org/pdf/1905.03222) | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n| [ConformalQRModel](https://unit8co.github.io/darts/generated_api/darts.models.forecasting.conformal_models.html#darts.models.forecasting.conformal_models.ConformalQRModel) | [Conformalized Quantile Regression](https://arxiv.org/pdf/1905.03222) | \u2705 \u2705 | \u2705 \u2705 \u2705 | \u2705 \u2705 | \u2705 |\n\n## Community & Contact\nAnyone is welcome to join our [Gitter room](https://gitter.im/u8darts/darts) to ask questions, make proposals,\ndiscuss use-cases, and more. If you spot a bug or have suggestions, GitHub issues are also welcome.\n\nIf what you want to tell us is not suitable for Gitter or Github,\nfeel free to send us an email at <a href=\"mailto:darts@unit8.co\">darts@unit8.co</a> for\ndarts related matters or <a href=\"mailto:info@unit8.co\">info@unit8.co</a> for any other\ninquiries.\n\n## Contribute\nThe development is ongoing, and we welcome suggestions, pull requests and issues on GitHub.\nAll contributors will be acknowledged on the\n[change log page](https://github.com/unit8co/darts/blob/master/CHANGELOG.md).\n\nBefore working on a contribution (a new feature or a fix),\n[check our contribution guidelines](https://github.com/unit8co/darts/blob/master/CONTRIBUTING.md).\n\n## Citation\nIf you are using Darts in your scientific work, we would appreciate citations to the following JMLR paper.\n\n[Darts: User-Friendly Modern Machine Learning for Time Series](https://www.jmlr.org/papers/v23/21-1177.html)\n\nBibtex entry:\n```\n@article{JMLR:v23:21-1177,\n author = {Julien Herzen and Francesco L\u00c3\u00a4ssig and Samuele Giuliano Piazzetta and Thomas Neuer and L\u00c3\u00a9o Tafti and Guillaume Raille and Tomas Van Pottelbergh and Marek Pasieka and Andrzej Skrodzki and Nicolas Huguenin and Maxime Dumonal and Jan Ko\u00c5\u203acisz and Dennis Bader and Fr\u00c3\u00a9d\u00c3\u00a9rick Gusset and Mounir Benheddi and Camila Williamson and Michal Kosinski and Matej Petrik and Ga\u00c3\u00abl Grosch},\n title = {Darts: User-Friendly Modern Machine Learning for Time Series},\n journal = {Journal of Machine Learning Research},\n year = {2022},\n volume = {23},\n number = {124},\n pages = {1-6},\n url = {http://jmlr.org/papers/v23/21-1177.html}\n}\n```\n",
"bugtrack_url": null,
"license": "Apache License 2.0",
"summary": "A python library for easy manipulation and forecasting of time series.",
"version": "0.32.0",
"project_urls": {
"Bug Tracker": "https://github.com/unit8co/darts/issues",
"Documentation": "https://unit8co.github.io/darts/",
"Homepage": "https://unit8co.github.io/darts/",
"Source Code": "https://github.com/unit8co/darts"
},
"split_keywords": [
"time",
"series",
"forecasting"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "57d1e8fabffd070e4b7444983e8d02d8a4ea686e658a29a90c368645f12a80b0",
"md5": "356f4e20a3fb6a2391fc10b59f787e04",
"sha256": "a066acd1790b76e22d69dcc69b19e8e4209174e85636432b59ff3288dcce9e30"
},
"downloads": -1,
"filename": "darts-0.32.0-py3-none-any.whl",
"has_sig": false,
"md5_digest": "356f4e20a3fb6a2391fc10b59f787e04",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.9",
"size": 963292,
"upload_time": "2024-12-21T13:55:32",
"upload_time_iso_8601": "2024-12-21T13:55:32.550787Z",
"url": "https://files.pythonhosted.org/packages/57/d1/e8fabffd070e4b7444983e8d02d8a4ea686e658a29a90c368645f12a80b0/darts-0.32.0-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "a8f27bde3ddcac7ed47c545ec725bfa28e654cd52ac7c683edb7dbe97fe603a0",
"md5": "7959b41921d8b6d43fe48cbcdc2c92c7",
"sha256": "1d56215bcbd9c9b6af11e0efb4018b958a925ced34fa4c2da3f27e5700f881a2"
},
"downloads": -1,
"filename": "darts-0.32.0.tar.gz",
"has_sig": false,
"md5_digest": "7959b41921d8b6d43fe48cbcdc2c92c7",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.9",
"size": 830851,
"upload_time": "2024-12-21T13:55:35",
"upload_time_iso_8601": "2024-12-21T13:55:35.438981Z",
"url": "https://files.pythonhosted.org/packages/a8/f2/7bde3ddcac7ed47c545ec725bfa28e654cd52ac7c683edb7dbe97fe603a0/darts-0.32.0.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-12-21 13:55:35",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "unit8co",
"github_project": "darts",
"travis_ci": false,
"coveralls": true,
"github_actions": true,
"lcname": "darts"
}