Darts: User-Friendly Modern Machine Learning for Time Series
About
We present Darts, a Python machine learning library for time series, with a focus on forecasting. Darts offers a variety of models, from classics such as ARIMA to state-of-the-art deep neural networks. The emphasis of the library is on offering modern machine learning functionalities, such as supporting multidimensional series, meta-learning on multiple series, training on large datasets, incorporating external data, ensembling models, and providing a rich support for probabilistic forecasting. At the same time, great care goes into the API design to make it user-friendly and easy to use. For instance, all models can be used using fit()/predict(), similar to scikit-learn.
Julien Herzen, Francesco L\"assig, Samuele Giuliano Piazzetta, Thomas Neuer, L\'eo Tafti, Guillaume Raille, Tomas Van Pottelbergh, Marek Pasieka, Andrzej Skrodzki, Nicolas Huguenin, Maxime Dumonal, Jan Ko\'scisz, Dennis Bader, Fr\'ed\'erick Gusset, Mounir Benheddi, Camila Williamson, Michal Kosinski, Matej Petrik, Ga\"el Grosch• 2021
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Long-term time-series forecasting | ETTh1 (test) | MSE0.952 | 221 | |
| Long-term time-series forecasting | Traffic (test) | MSE1.344 | 116 | |
| Long-term time-series forecasting | Weather (test) | MSE0.477 | 103 | |
| Long-term time-series forecasting | ETTh2 (test) | MSE0.635 | 92 | |
| Long-term time-series forecasting | ETTm1 (test) | MSE1.145 | 81 | |
| Long-term forecasting | Electricity (test) | MSE0.597 | 79 | |
| Long-term forecasting | ILI (test) | MSE4.323 | 20 | |
| Long-term time-series forecasting | ETTm2 (test) | MSE0.975 | 17 | |
| Vegetation Forecasting | GreenEarthNet OOD-t 100-day lead time | R20.41 | 14 | |
| Vegetation Forecasting | GreenEarthNet 25-day lead time (OOD-t) | RMSE0.16 | 14 |
Showing 10 of 11 rows