time-series-foundation-models / lag-llama

Lag-Llama: Towards Foundation Models for Probabilistic Time Series Forecasting
Apache License 2.0
1.08k stars 121 forks source link

Autogluon support #71

Open bvrabete opened 3 weeks ago

bvrabete commented 3 weeks ago

AutoGluon is an AutoML framework that also offers support for time series forecasting. Most if not all traditional models are supported (Arima, DeepAR) but also new models such as PatchTST and Chronos.

The useful part about that is that it makes fine tuning and comparison versy easy.

Would it b possible to enable AutoGluon support for your model? Or point out how a wrapper class for Lag LLama could be written?

ashok-arjun commented 2 weeks ago

Hi! Thanks for the issue.

We do not have plans to integrate our model with AutoGluon at the moment. I am not sure how a wrapper class can be written too, but I think it shouldn't be very difficult given we use the GluonTS backend.

Maybe @kashif has ideas on the best ways to integrate Lag-Llama into AutoGluon.