Nixtla / statsforecast

Lightning ⚡️ fast forecasting with statistical and econometric models.
https://nixtlaverse.nixtla.io/statsforecast
Apache License 2.0
3.92k stars 276 forks source link

[Models] Add possibility to ensemble different models #520

Open AzulGarza opened 1 year ago

AzulGarza commented 1 year ago

Description

Often, some pipelines require the ensembling of different statistical models. We can accomplish this by introducing a new module in the library named statsforecast/ensembles.py, which contains various methods for ensembling models. The general implementation might look something like this:

from statsforecast.ensembles import FFORMA, MedianEnsemble

sf = StatsForecast(
     models=[FFORMA(AutoARIMA(), AutoETS()), MedianEnsemble(Naive(), SeasonalNaive())]
)

Ideally, the output dataframe should include the forecasts from individual models as well as the ensemble method's forecast.

Use case

No response

hberande commented 1 month ago

Hi @AzulGarza, have we implemented above ensemble model in NIXTLA ?

anthonygiorgio97 commented 1 month ago

Hi @AzulGarza . It seems to be a really interesting solution. Are there any plans to implement it in the short term?