From bd2a646e1af4b2f6f2f95802ae6dbcb01345af52 Mon Sep 17 00:00:00 2001 From: Garrett Wu Date: Mon, 25 Mar 2024 21:05:04 +0000 Subject: [PATCH] docs: fix docs of ARIMAPlus.predict --- bigframes/ml/forecasting.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/bigframes/ml/forecasting.py b/bigframes/ml/forecasting.py index 292389dcbb..7993327200 100644 --- 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):