@@ -51,6 +51,34 @@ def create_model(
5151) -> bigframes .ml .base .BaseEstimator :
5252 """
5353 Creates a BigQuery ML model.
54+
55+ Args:
56+ model_name (str):
57+ The name of the model in BigQuery.
58+ replace (bool, default False):
59+ Whether to replace the model if it already exists.
60+ if_not_exists (bool, default False):
61+ Whether to ignore the error if the model already exists.
62+ transform (list[str], optional):
63+ The TRANSFORM clause, which specifies the preprocessing steps to apply to the input data.
64+ input_schema (Mapping[str, str], optional):
65+ The INPUT clause, which specifies the schema of the input data.
66+ output_schema (Mapping[str, str], optional):
67+ The OUTPUT clause, which specifies the schema of the output data.
68+ connection_name (str, optional):
69+ The connection to use for the model.
70+ options (Mapping[str, Union[str, int, float, bool, list]], optional):
71+ The OPTIONS clause, which specifies the model options.
72+ training_data (Union[dataframe.DataFrame, str], optional):
73+ The query or DataFrame to use for training the model.
74+ custom_holiday (Union[dataframe.DataFrame, str], optional):
75+ The query or DataFrame to use for custom holiday data.
76+ session (bigframes.session.Session, optional):
77+ The BigFrames session to use. If not provided, the default session is used.
78+
79+ Returns:
80+ bigframes.ml.base.BaseEstimator:
81+ The created BigFrames model.
5482 """
5583 import bigframes .pandas as bpd
5684
0 commit comments