Skip to content
This repository was archived by the owner on Apr 1, 2026. It is now read-only.

Commit 9231f56

Browse files
committed
restore docstring from jules
1 parent 7df5e09 commit 9231f56

1 file changed

Lines changed: 28 additions & 0 deletions

File tree

  • bigframes/bigquery/_operations

bigframes/bigquery/_operations/ml.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)