docs: fix docs of ARIMAPlus.predict (#512)

googleapis · Mar 26, 2024 · 3b80f95 · 3b80f95
1 parent 6873b30
commit 3b80f95
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bigframes/ml/forecasting.py b/bigframes/ml/forecasting.py
@@ -234,7 +234,12 @@ def _fit(
     def predict(
         self, X=None, *, horizon: int = 3, confidence_level: float = 0.95
     ) -> bpd.DataFrame:
-        """Predict the closest cluster for each sample in X.
+        """Forecast time series at future horizon.
+
+        .. note::
+
+            Output matches that of the BigQuery ML.FORECAST function.
+            See: https://cloud.google.com/bigquery/docs/reference/standard-sql/bigqueryml-syntax-forecast
 
         Args:
             X (default None):