Fix SARIMAX seasonal_order parameter (Significant-Gravitas#711)

SquareandCompass · Sep 6, 2022 · 87d9b35 · 87d9b35
1 parent 9901156
commit 87d9b35
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flaml/model.py b/flaml/model.py
@@ -1975,7 +1975,7 @@ def fit(self, X_train, y_train, budget=None, **kwargs):
                 train_df[[TS_VALUE_COL]],
                 exog=train_df[regressors],
                 order=(self.params["p"], self.params["d"], self.params["q"]),
-                seasonality_order=(
+                seasonal_order=(
                     self.params["P"],
                     self.params["D"],
                     self.params["Q"],
@@ -1988,7 +1988,7 @@ def fit(self, X_train, y_train, budget=None, **kwargs):
             model = SARIMAX_estimator(
                 train_df,
                 order=(self.params["p"], self.params["d"], self.params["q"]),
-                seasonality_order=(
+                seasonal_order=(
                     self.params["P"],
                     self.params["D"],
                     self.params["Q"],