From 63625b41c8bb168d27632156d74e386ba9d81689 Mon Sep 17 00:00:00 2001 From: Adam Dupaski Date: Fri, 8 Aug 2025 10:34:01 +0200 Subject: [PATCH] Update quickstart.py --- samples/snippets/quickstart.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/samples/snippets/quickstart.py b/samples/snippets/quickstart.py index bc05cd2512..08662c1ea7 100644 --- a/samples/snippets/quickstart.py +++ b/samples/snippets/quickstart.py @@ -16,7 +16,7 @@ def run_quickstart(project_id: str) -> None: your_gcp_project_id = project_id - # [START bigquery_bigframes_quickstart] + # [START bigquery_bigframes_quickstart_create_dataframe] import bigframes.pandas as bpd # Set BigQuery DataFrames options @@ -37,12 +37,16 @@ def run_quickstart(project_id: str) -> None: # Efficiently preview the results using the .peek() method. df.peek() + # [END bigquery_bigframes_quickstart_create_dataframe] + # [START bigquery_bigframes_quickstart_calculate_print] # Use the DataFrame just as you would a pandas DataFrame, but calculations # happen in the BigQuery query engine instead of the local system. average_body_mass = df["body_mass_g"].mean() print(f"average_body_mass: {average_body_mass}") + # [END bigquery_bigframes_quickstart_calculate_print] + # [START bigquery_bigframes_quickstart_eval_metrics] # Create the Linear Regression model from bigframes.ml.linear_model import LinearRegression @@ -70,7 +74,7 @@ def run_quickstart(project_id: str) -> None: model = LinearRegression(fit_intercept=False) model.fit(X, y) model.score(X, y) - # [END bigquery_bigframes_quickstart] + # [END bigquery_bigframes_quickstart_eval_metrics] # close session and reset option so not to affect other tests bpd.close_session()