Create a clustering model with BigQuery DataFrames
Stay organized with collections
Save and categorize content based on your preferences.
Create a k-means clustering model on the lengths and sex of penguins using the BigQuery DataFrames API.
Explore further
For detailed documentation that includes this code sample, see the following:
Code sample
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License
, and code samples are licensed under the Apache 2.0 License
. For details, see the Google Developers Site Policies
. Java is a registered trademark of Oracle and/or its affiliates.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Hard to understand","hardToUnderstand","thumb-down"],["Incorrect information or sample code","incorrectInformationOrSampleCode","thumb-down"],["Missing the information/samples I need","missingTheInformationSamplesINeed","thumb-down"],["Other","otherDown","thumb-down"]],[],[[["\u003cp\u003eThis content demonstrates creating a k-means clustering model using the BigQuery DataFrames API.\u003c/p\u003e\n"],["\u003cp\u003eThe model is trained on penguin data, specifically using culmen length and sex as features.\u003c/p\u003e\n"],["\u003cp\u003eThe example code showcases how to load data from BigQuery, create and fit the KMeans model, and then predict and score it.\u003c/p\u003e\n"],["\u003cp\u003eYou can find more detailed documentation on using BigQuery DataFrames, setting up authentication, and using the BigQuery Python API in the provided links.\u003c/p\u003e\n"]]],[],null,["# Create a clustering model with BigQuery DataFrames\n\nCreate a k-means clustering model on the lengths and sex of penguins using the BigQuery DataFrames API.\n\nExplore further\n---------------\n\n\nFor detailed documentation that includes this code sample, see the following:\n\n- [Use BigQuery DataFrames](/bigquery/docs/use-bigquery-dataframes)\n\nCode sample\n-----------\n\n### Python\n\n\nBefore trying this sample, follow the Python setup instructions in the\n[BigQuery quickstart using\nclient libraries](/bigquery/docs/quickstarts/quickstart-client-libraries).\n\n\nFor more information, see the\n[BigQuery Python API\nreference documentation](/python/docs/reference/bigquery/latest).\n\n\nTo authenticate to BigQuery, set up Application Default Credentials.\nFor more information, see\n\n[Set up authentication for client libraries](/bigquery/docs/authentication#client-libs).\n\n from bigframes.ml.cluster import KMeans\n import bigframes.pandas as bpd\n\n # Load data from BigQuery\n query_or_table = \"bigquery-public-data.ml_datasets.penguins\"\n bq_df = bpd.read_gbq(query_or_table)\n\n # Create the KMeans model\n cluster_model = KMeans(n_clusters=10)\n cluster_model.fit(bq_df[\"culmen_length_mm\"], bq_df[\"sex\"])\n\n # Predict using the model\n result = cluster_model.predict(bq_df)\n # Score the model\n score = cluster_model.score(bq_df)\n\nWhat's next\n-----------\n\n\nTo search and filter code samples for other Google Cloud products, see the\n[Google Cloud sample browser](/docs/samples?product=bigquery)."]]
