-
Notifications
You must be signed in to change notification settings - Fork 80
Create MultisegmentModel #551
Conversation
Codecov Report
@@ Coverage Diff @@
## master #551 +/- ##
==========================================
- Coverage 87.34% 87.27% -0.08%
==========================================
Files 118 118
Lines 5704 5688 -16
==========================================
- Hits 4982 4964 -18
- Misses 722 724 +2
Continue to review full report at Codecov.
|
etna/models/base.py
Outdated
@@ -336,3 +336,72 @@ def forecast( | |||
ts.df = df | |||
ts.inverse_transform() | |||
return ts | |||
|
|||
|
|||
class MultisegmentModel(FitAbstractModel, ForecastAbstractModel, BaseMixin): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
MultisegmentModel
vs PerSegmentBaseModel
, PerSegmentModel
-- in the first case we use lower s
, in the second -- upper S
etna/models/base.py
Outdated
Model after fit | ||
""" | ||
df = ts.to_pandas(flatten=True) | ||
df = df.dropna() # TODO |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can we create issue and reference it here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
👍
IMPORTANT: Please do not create a Pull Request without creating an issue first.
Before submitting (must do checklist)
Type of Change
Proposed Changes
Related Issue
Closing issues
closes #509