Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

TaskDatasetResultRank
Long-term time-series forecastingETTh1 (test)
MSE0.952
221
Long-term time-series forecastingTraffic (test)
MSE1.344
116
Long-term time-series forecastingWeather (test)
MSE0.477
103
Long-term time-series forecastingETTh2 (test)
MSE0.635
92
Long-term time-series forecastingETTm1 (test)
MSE1.145
81
Long-term forecastingElectricity (test)
MSE0.597
79
Long-term forecastingILI (test)
MSE4.323
20
Long-term time-series forecastingETTm2 (test)
MSE0.975
17
Vegetation ForecastingGreenEarthNet OOD-t 100-day lead time
R20.41
14
Vegetation ForecastingGreenEarthNet 25-day lead time (OOD-t)
RMSE0.16
14
Showing 10 of 11 rows

Other info

Follow for update