Share your thoughts, 1 month free Claude Pro on usSee more
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
264
Long-term time-series forecastingTraffic (test)
MSE1.344
149
Long-term time-series forecastingWeather (test)
MSE0.477
147
Long-term time-series forecastingETTm1 (test)
MSE1.145
138
Long-term time-series forecastingETTh2 (test)
MSE0.635
135
Long-term forecastingElectricity (test)
MSE0.597
79
Long-term time-series forecastingETTm2 (test)
MSE0.975
61
Long-term forecastingILI (test)--
24
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