tinkoff-ai / etna

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

Add `params_to_tune` for exponential smoothing models #1195

Closed Mr-Geekman closed 1 year ago

Mr-Geekman commented 1 year ago

🚀 Feature Request

Add params_to_tune for exponential smoothing models.

Proposal

Sources:

Suggested grid for HoltWintersModel:

{
    "trend": CategoricalDistribution(["add", "mul", None]),
    "damped_trend": CategoricalDistribution([False, True]),
    "seasonal": CategoricalDistribution(["add", "mul", None]),
    "use_boxcox": CategoricalDistribution([False, True]),
}

Suggested grid for HoltModel:

{
    "exponential": CategoricalDistribution([False, True]),
    "damped_trend": CategoricalDistribution([False, True]),
    "use_boxcox": CategoricalDistribution([False, True]),
}

Suggested grid for SimpleExpSmoothingModel:

{
    "use_boxcox": CategoricalDistribution([False, True]),
}

To discuss:

Test cases

Look at https://github.com/tinkoff-ai/etna/issues/1184.

Additional context

No response

martins0n commented 1 year ago
Mr-Geekman commented 1 year ago

Closed by #1209.