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