Use the Data Science Agent Skip to main content
Stay organized with collections Save and categorize content based on your preferences.
Use the Data Science AgentPreview
This feature is subject to the "Pre-GA Offerings Terms" in the General Service Terms section of the Service Specific Terms. Pre-GA features are available "as is" and might have limited support. For more information, see the launch stage descriptions.
Note: To provide feedback or ask questions about this Preview feature, contact vertex-notebooks-previews-external@google.com.This guide describes how you can use the Data Science Agent in Colab Enterprise to help you perform data science tasks in your notebooks.
Learn how and when Gemini for Google Cloud uses your data.
This document is intended for data analysts, data scientists, and data developers who work with Colab Enterprise. It assumes you have knowledge of how to write code in a notebook environment.
Capabilities of the Data Science AgentThe Data Science Agent can help you with tasks ranging from exploratory data analysis to generating machine learning predictions and forecasts. You can use the Data Science Agent for:
DecisionTreeRegressor
and RandomForestRegressor
and compare their performance.In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Note: If you don't plan to keep the resources that you create in this procedure, create a project instead of selecting an existing project. After you finish these steps, you can delete the project, removing all resources associated with the project.Verify that billing is enabled for your Google Cloud project.
Enable the Vertex AI, Dataform, and Compute Engine APIs.
In the Google Cloud console, on the project selector page, select or create a Google Cloud project.
Note: If you don't plan to keep the resources that you create in this procedure, create a project instead of selecting an existing project. After you finish these steps, you can delete the project, removing all resources associated with the project.Verify that billing is enabled for your Google Cloud project.
Enable the Vertex AI, Dataform, and Compute Engine APIs.
To get the permissions that you need to use the Data Science Agent in Colab Enterprise, ask your administrator to grant you the Colab Enterprise User (roles/aiplatform.colabEnterpriseUser
) IAM role on the project. For more information about granting roles, see Manage access to projects, folders, and organizations.
You might also be able to get the required permissions through custom roles or other predefined roles.
One or more of the required roles includes thedataform.repositories.list
permission. Users who are granted the dataform.repositories.list
permission or the Code Creator (roles/dataform.codeCreator
) role in a project can list the names of code assets in that project by using the Dataform API or the Dataform command-line interface (CLI). Non-administrators using BigQuery Studio can only see code assets that they created or that were shared with them. Use the Data Science Agent
To get started using Colab Enterprise's Data Science Agent, do the following:
In the Google Cloud console, go to the Colab Enterprise My notebooks page.
In the Region menu, select the region that contains your notebook.
Click the notebook that you want to open.
In the toolbar, click the spark Gemini button to open the chat dialog.
Note: You can move the chat dialog into a separate panel outside the notebook by clicking the dock_to_left Move to panel button.To upload a CSV file, do the following:
If necessary, authorize your Google Account.
Wait a moment for Colab Enterprise to start a runtime and enable file browsing.
Click OK to acknowledge that this runtime's files will be deleted when the runtime is deleted.
The file is uploaded to the Files pane.
Next to the file that you uploaded, click the more_vertActions menu, and then select Add to Gemini.
The file is added to the chat dialog.
In the Gemini chat dialog, enter a prompt and click send Send. To get ideas for prompts, review the [Data Science Agent capabilities](#capabilities) and see [Sample prompts](#sample-prompts).
For example, you might enter "Provide an analysis of the data I've uploaded."
Gemini responds to your prompt. The response can include code snippets to run, general advice for your project, next steps for accomplishing your goals, or information about specific problems in your data or code.
After evaluating the response, you can do the following:
To close the Gemini dialog, click close Close.
To turn off Gemini in Colab Enterprise for a Google Cloud project, an administrator must turn off the Gemini for Google Cloud API. See Disabling services.
To turn off Gemini in Colab Enterprise for a specific user, an administrator needs to revoke the Gemini for Google Cloud User (roles/cloudaicompanion.user
) role for that user. See Revoke a single IAM role.
The following examples show the types of prompts that you can use with the Data Science Agent.
experience_level
column to group the salaries, and create a box plot for each group showing the values from the salary_in_usd
column.class
variable of a specific fruit. Split the data into training and test datasets to generate a model and then evaluate the model's accuracy. Create a confusion matrix to show the predictions for each class, including all predictions that are correct and incorrect.target_variable
from filename.csv
for the next six months.filename.csv
for target_variable
.To view the supported regions for Colab Enterprise's Data Science Agent, see Locations.
BillingDuring Preview, you are charged only for running code in the notebook's runtime. For more information, see Colab Enterprise pricing.
What's nextFor information on how to use the Data Science Agent with BigQuery, see Use the Colab Enterprise Data Science Agent with BigQuery.
Read the Gemini for Google Cloud overview.
For more ways to write and edit code with Gemini assistance, see the following:
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.
Last updated 2025-08-07 UTC.
[[["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"]],["Last updated 2025-08-07 UTC."],[],[]]
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4