From c092fba0066e2cdb8a8362791e5d2fb1183f8329 Mon Sep 17 00:00:00 2001 From: Bobo Jamson <82302227+bobo-jamson@users.noreply.github.com> Date: Tue, 23 Apr 2024 20:22:33 -0500 Subject: [PATCH] magic number 5 replaced with self.window --- src/forecasting/baselines.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/forecasting/baselines.py b/src/forecasting/baselines.py index 8cd823e..c850fb1 100644 --- a/src/forecasting/baselines.py +++ b/src/forecasting/baselines.py @@ -21,7 +21,7 @@ def fit(self, series: TimeSeries): super().fit(series) self._history = series.data_array().to_series().values self._fitted_values = ( - series.data_array().to_series().rolling(window=5).mean().bfill() + series.data_array().to_series().rolling(window=self.window).mean().bfill() ) def predict(self, n: int, num_samples: int = 1): @@ -35,4 +35,4 @@ def predict(self, n: int, num_samples: int = 1): return self._build_forecast_series(forecast) def supports_multivariate(self) -> bool: - return False \ No newline at end of file + return False