You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Create _TBATSAdapter inherit it from BaseAdapter. It should implement methods fit, predict and get_model.
Also create _TBATSPerSegmentModel(PerSegmentPredictionIntervalModel) as a model it should expect class BATS or TBATS from tbats (also you may use Estimator).
After that create classes BATSPerSegmentModel and TBATSPerSegmentModel
🚀 Feature Request
Add new models to library.
Motivation
Should expand number of models in ETNA
Proposal
Add BATS and TBATS model adapters (use models from tbats library https://github.com/intive-DataScience/tbats)
Create
_TBATSAdapter
inherit it from BaseAdapter. It should implement methodsfit
,predict
andget_model
.Also create
_TBATSPerSegmentModel(PerSegmentPredictionIntervalModel)
as a model it should expect classBATS
orTBATS
from tbats (also you may use Estimator).After that create classes
BATSPerSegmentModel
andTBATSPerSegmentModel
To understand the idea look at the
etna/etna/models/sklearn.py
Line 13 in be1b719
etna/etna/models/sklearn.py
Line 75 in be1b719
etna/etna/models/linear.py
Line 8 in be1b719
Test cases
etna/tests/test_models/test_linear_model.py
Lines 100 to 107 in be1b719
Alternatives
No response
Additional context
Look at the sktime for inspiration https://github.com/alan-turing-institute/sktime/blob/2a0df3f52a6fe67f5509f6199cc60f01e3dc0adf/sktime/forecasting/tbats.py#L15
Checklist
The text was updated successfully, but these errors were encountered: