tinkoff-ai / etna

ETNA – Time-Series Library

Home Page:https://etna.tinkoff.ru

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

`CrostonModel`

Mr-Geekman opened this issue · comments

🚀 Feature Request

Add model based on Croston's method for intermittent demand.

We could also try to implement some variations of this method.

Proposal

Add CrostonModel.

  • It could be implemeted as per-segment model using BaseAdapter and PerSegmentModelMixin (look at ProphetModel for reference)
  • It could be implemented as NonPredictionIntervalContextIgnorantAbstractModel that saves the answer during the fit
  • During fit it should raise a warning that it doesn't work with exogenous features, look at SeasonalMovingAverageModel

References:

To discuss:

  • Should it be our own implementation or some external package can be used (statsforecast for example)
    • We should try to make our own implementation

Test cases

  • Tests for the model itself
  • Inference tests in tests/test_models/test_inference/.

Additional context

No response