@@ -52,6 +52,10 @@ def create_model(
5252 """
5353 Creates a BigQuery ML model.
5454
55+ See the `BigQuery ML CREATE MODEL DDL syntax
56+ <https://docs.cloud.google.com/bigquery/docs/reference/standard-sql/bigqueryml-syntax-create>`_
57+ for additional reference.
58+
5559 Args:
5660 model_name (str):
5761 The name of the model in BigQuery.
@@ -69,16 +73,16 @@ def create_model(
6973 The connection to use for the model.
7074 options (Mapping[str, Union[str, int, float, bool, list]], optional):
7175 The OPTIONS clause, which specifies the model options.
72- training_data (Union[dataframe .DataFrame, str], optional):
76+ training_data (Union[bigframes.pandas .DataFrame, str], optional):
7377 The query or DataFrame to use for training the model.
74- custom_holiday (Union[dataframe .DataFrame, str], optional):
78+ custom_holiday (Union[bigframes.pandas .DataFrame, str], optional):
7579 The query or DataFrame to use for custom holiday data.
7680 session (bigframes.session.Session, optional):
77- The BigFrames session to use. If not provided, the default session is used.
81+ The session to use. If not provided, the default session is used.
7882
7983 Returns:
8084 bigframes.ml.base.BaseEstimator:
81- The created BigFrames model.
85+ The created BigQuery ML model.
8286 """
8387 import bigframes .pandas as bpd
8488
0 commit comments