From 6d894fcd8bf4cb036922767a2ef8a400267da7db Mon Sep 17 00:00:00 2001 From: Julia Shenshina Date: Fri, 8 Oct 2021 08:55:28 +0300 Subject: [PATCH] Fix --- etna/ensembles/voting_ensemble.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/etna/ensembles/voting_ensemble.py b/etna/ensembles/voting_ensemble.py index 13f61d170..8f2957bcf 100644 --- a/etna/ensembles/voting_ensemble.py +++ b/etna/ensembles/voting_ensemble.py @@ -124,9 +124,7 @@ def _forecast_pipeline(pipeline: Pipeline) -> TSDataset: def _vote(self, forecasts: List[TSDataset]) -> TSDataset: """Get average forecast.""" - forecast_df = forecasts[0][:, :, "target"] * self.weights[0] - for forecast, weight in zip(forecasts[1:], self.weights[1:]): - forecast_df += forecast[:, :, "target"] * weight + forecast_df = sum([forecast[:, :, "target"] * weight for forecast, weight in zip(forecasts, self.weights)]) forecast_dataset = TSDataset(df=forecast_df, freq=forecasts[0].freq) return forecast_dataset