From 9f8bed64bc1bc5a1bf1531e4868997b8c632835c Mon Sep 17 00:00:00 2001 From: Tony Bagnall Date: Tue, 5 Nov 2024 20:19:03 +0000 Subject: [PATCH] _forecast --- aeon/forecasting/base.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/aeon/forecasting/base.py b/aeon/forecasting/base.py index b7cd791613..023e9d6db7 100644 --- a/aeon/forecasting/base.py +++ b/aeon/forecasting/base.py @@ -109,7 +109,13 @@ def forecast(self, y, X=None): single prediction directly after the last point in X. """ y = self._preprocess_series(y, axis=self.axis, store_metadata=False) + self.is_fitted = True return self._forecast(y, X) - @abstractmethod - def _forecast(self, y=None, exog=None): ... + def _forecast(self, y=None, exog=None): + """Forecast values for time series X. + + NOTE: deal with horizons + """ + self.fit(y, exog) + return self.predict(y, exog)