diff --git a/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/statsmodels.py b/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/statsmodels.py index b2ebced32c..e8fceb1531 100644 --- a/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/statsmodels.py +++ b/fedot/core/operations/evaluation/operation_implementations/models/ts_implementations/statsmodels.py @@ -295,7 +295,7 @@ def predict(self, input_data): predictions = self.model.forecast(steps=forecast_length) predict = np.array(predictions).reshape(1, -1) - input_data.idx = np.arange(start_id, end_id) + input_data.idx = np.arange(start_id, end_id + 1) output_data = self._convert_to_output(input_data, predict=predict, @@ -345,4 +345,10 @@ def _check_and_correct_params(self, endog: np.ndarray) -> bool: self.params.update(**{'trend': 'add'}) params_changed = True + if self.params.get('seasonal'): + self.seasonal_periods = min(int(0.5 * (len(endog) - 1)), self.seasonal_periods) + self.seasonal_periods = max(self.seasonal_periods, 1) + self.params.update(**{'seasonal_periods': self.seasonal_periods}) + params_changed = True + return params_changed