-
Notifications
You must be signed in to change notification settings - Fork 80
Add STLTransform
#158
Add STLTransform
#158
Conversation
# Conflicts: # etna/transforms/__init__.py
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.
🔥
etna/transforms/stl.py
Outdated
stl_kwargs: Optional[Dict[str, Any]] = None, | ||
): | ||
""" | ||
Init _OneSegmentSTLTransform. |
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.
Init _OneSegmentSTLTransform. | |
Init STLTransform. |
if isinstance(model, str): | ||
if model == "arima": | ||
self.model = ARIMA | ||
model_kwargs = {"order": (1, 1, 0)} |
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.
Different default verisions in docstring and this line.
raise ValueError(f"Not a valid option for model: {model}") | ||
elif isinstance(model, TimeSeriesModel): | ||
self.model = model | ||
|
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.
I guess we shoud add else
block too
Codecov Report
@@ Coverage Diff @@
## master #158 +/- ##
==========================================
+ Coverage 87.01% 87.12% +0.11%
==========================================
Files 73 74 +1
Lines 3180 3239 +59
==========================================
+ Hits 2767 2822 +55
- Misses 413 417 +4
Continue to review full report at Codecov.
|
IMPORTANT: Please do not create a Pull Request without creating an issue first.
Before submitting (must do checklist)
Type of Change
Proposed Changes
Add
STLTransform
class that subtracts trend and seasonal components.Related Issue
#130.
Closing issues
Closes #130.