tinkoff-ai / etna

ETNA – Time-Series Library
https://etna.tinkoff.ru
Apache License 2.0
862 stars 80 forks source link

Add `params_to_tune` for arima models #1194

Closed Mr-Geekman closed 1 year ago

Mr-Geekman commented 1 year ago

🚀 Feature Request

Add params_to_tune for arima models.

Proposal

Sources for the grid:

Suggested grid for SARIMAXModel:

{
    "p": IntUniformDistribution(low=1, high=6, step=1),
    "d": IntUniformDistribution(low=1, high=3, step=1),
    "q": IntUniformDistribution(low=1, high=6, step=1),
    "P": IntUniformDistribution(low=0, high=2, step=1),
    "D": IntUniformDistribution(low=0, high=1, step=1),
    "Q": IntUniformDistribution(low=0, high=1, step=1),
    "trend": CategoricalDistribution(["n", "c", "t", "ct"]),
}

Suggested grid for AutoARIMAModel:

{}

To discuss:

Test cases

Look at #1184.

Additional context

Is blocked by #1201.

Mr-Geekman commented 1 year ago

Closed by #1206.