etna
etna copied to clipboard
Add aggregation method parameter to Imputers
🚀 Feature Request
Add possibility to perform different kids of aggregation for seasonal
strategy in TimeSeriesImputerTransform
Proposal
- In
TimeSeriesImputerTransform
add parameteraggregation_method: AggregationMode = "AggregationMode.mean
-- method to aggregate values in the window (see link) - Update method
_fill
to handle the logic withaggregation_method
-- now it should aggregate values according the the passed method https://github.com/tinkoff-ai/etna/blob/cfbfb01d28fd330ca42de7a7176384708174920d/etna/transforms/missing_values/imputation.py#L203 - Update the docs
Test cases
Add test for the seasonal
strategy which checks that it works correctly with different aggregation_method
Additional context
No response