All Vertex AI code samples

This page contains code samples for Vertex AI. To search and filter code samples for other Google Cloud products, see the Google Cloud sample browser.

Cancel a batch prediction job

Cancels a batch prediction job using the cancel_batch_prediction_job method.

Java Node.js Python

View sample

Cancel a custom job

Cancels a custom job using the cancel_custom_job method.

Node.js Python

View sample

Cancel a data labeling job

Cancels a data labeling job using the cancel_data_labeling_job method.

Java Python

View sample

Cancel a hyperparameter tuning job

Cancels a hyperparameter tuning job using the cancel_hyperparameter_tuning_job method.

Python

View sample

Cancel a training pipeline

Cancels a training pipeline using the cancel_training_pipeline method.

Java Python

View sample

Create a batch prediction job

Creates a batch prediction job using the create_batch_prediction_job method.

Java Python

View sample

View in documentation

• Get predictions from a custom trained model

Create a batch prediction job for BigQuery

Creates a batch prediction job for BigQuery using the create_batch_prediction_job method.

Java Python

View sample

View in documentation

• Get batch predictions and explanations
• Get predictions for a forecast model
• Get TabNet batch predictions
• Get Wide & Deep batch predictions

Create a batch prediction job for tabular forecasting

Creates a batch prediction job for tabular forecasting using the create_batch_prediction_job method.

Python

View sample

Create a batch prediction job for text classification

Creates a batch prediction job for text classification using the create_batch_prediction_job method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a text classification model

Create a batch prediction job for text entity extraction

Creates a batch prediction job for text entity extraction using the create_batch_prediction_job method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a text entity extraction model

Create a batch prediction job for text sentiment analysis

Creates a batch prediction job for text sentiment analysis using the create_batch_prediction_job method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a text sentiment analysis model

Create a batch prediction job for video action recognition

Creates a batch prediction job for video action recognition using the create_batch_prediction_job method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a video action recognition model

Create a batch prediction job for video classification

Creates a batch prediction job for video classification using the create_batch_prediction_job method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a video classification model

Create a batch prediction job for video object tracking

Creates a batch prediction job for video object tracking using the create_batch_prediction_job method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a video object tracking model

Create a custom job

Creates a custom job using the create_custom_job method.

Java Node.js Python

View sample

View in documentation

• Configure compute resources for custom training
• Configure container settings for custom training
• Create custom training jobs

Create a data labeling job

Creates a data labeling job using the create_data_labeling_job method.

Java Python

View sample

View in documentation

• Request Vertex AI data labeling

Create a data labeling job for active learning

Creates a data labeling job for active learning using the create_data_labeling_job method.

Java Python

View sample

Create a data labeling job for image segmentation

Creates a data labeling job for image segmentation using the create_data_labeling_job method.

Python

View sample

Create a data labeling job for images

Creates a data labeling job for images using the create_data_labeling_job method.

Python

View sample

Create a data labeling job for specialist pool

Creates a data labeling job for specialist pool using the create_data_labeling_job method.

Java Python

View sample

Create a data labeling job for video

Creates a data labeling job for video using the create_data_labeling_job method.

Java Python

View sample

Create a dataset

Creates a dataset using the create_dataset method.

Java Node.js Python

View sample

Create a dataset for image

Creates a dataset for image using the create_dataset method.

Java Node.js Python Terraform

View sample

View in documentation

• Create a dataset for training image classification models
• Create a dataset for training image object detection models

Create a dataset for tabular BigQuery

Creates a dataset for tabular BigQuery using the create_dataset method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training classification and regression models
• Create a dataset for training forecast models

Create a dataset for tabular Cloud Storage

Creates a dataset for tabular Cloud Storage using the create_dataset method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training classification and regression models
• Create a dataset for training forecast models

Create a dataset for text

Creates a dataset for text using the create_dataset method.

Java Node.js Python Terraform

View sample

View in documentation

• Create a dataset for training text classification models
• Create a dataset for training text entity extraction models
• Create a dataset for training text sentiment analysis models

Create a dataset for video

Creates a dataset for video using the create_dataset method.

Java Node.js Python Terraform

View sample

View in documentation

• Create a dataset for training video action recognition models
• Create a dataset for training video classification models
• Create a dataset for training video object tracking models

Create a hyperparameter tuning job

Creates a hyperparameter tuning job using the create_hyperparameter_tuning_job method.

Java Node.js Python

View sample

Create a hyperparameter tuning job for python package

Creates a hyperparameter tuning job for python package using the create_hyperparameter_tuning_job method.

Java Python

View sample

View in documentation

• Create a hyperparameter tuning job

Create a training pipeline

Creates a training pipeline using the create_training_pipeline method.

Java Python

View sample

Create a training pipeline for custom job

Creates a training pipeline for custom job using the create_training_pipeline method.

Java Python

View sample

View in documentation

• Create training pipelines

Create a training pipeline for custom training managed dataset

Creates a training pipeline for custom training managed dataset using the create_training_pipeline method.

Java Python

View sample

Create a training pipeline for image classification

Creates a training pipeline for image classification using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train an image classification model

Create a training pipeline for image object detection

Creates a training pipeline for image object detection using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train an image object detection model

Create a training pipeline for tabular classification

Creates a training pipeline for tabular classification using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a classification or regression model

Create a training pipeline for tabular forecasting

Creates a training pipeline for tabular forecasting using the create_training_pipeline method.

Python

View sample

Create a training pipeline for tabular regression

Creates a training pipeline for tabular regression using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a classification or regression model

Create a training pipeline for text classification

Creates a training pipeline for text classification using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a text classification model

Create a training pipeline for text entity extraction

Creates a training pipeline for text entity extraction using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a text entity extraction model

Create a training pipeline for text sentiment analysis

Creates a training pipeline for text sentiment analysis using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a text sentiment analysis model

Create a training pipeline for video action recognition

Creates a training pipeline for video action recognition using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a video action recognition model

Create a training pipeline for video classification

Creates a training pipeline for video classification using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a video classification model

Create a training pipeline for video object tracking

Creates a training pipeline for video object tracking using the create_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train a video object tracking model

Create an endpoint

Creates an endpoint using the create_endpoint method.

Java Node.js Python Terraform

View sample

View in documentation

• Get online predictions and explanations
• Get predictions from a custom trained model
• Get predictions from a image classification model
• Get predictions from a image object detection model
• Get predictions from a text classification model
• Get predictions from a text entity extraction model
• Get predictions from a text sentiment analysis model
• Get TabNet online predictions
• Get Wide & Deep online predictions

Delete a batch prediction job

Deletes a batch prediction job using the delete_batch_prediction_job method.

Java Node.js Python

View sample

Delete a custom job

Deletes a custom job using the delete_custom_job method.

Node.js Python

View sample

Delete a data labeling job

Deletes a data labeling job using the delete_data_labeling_job method.

Java Python

View sample

Delete a dataset

Deletes a dataset using the delete_dataset method.

Java Node.js Python

View sample

Delete a hyperparameter tuning job

Deletes a hyperparameter tuning job using the delete_hyperparameter_tuning_job method.

Python

View sample

Delete a model

Deletes a model using the delete_model method.

Java Node.js Python

View sample

Delete a training pipeline

Deletes a training pipeline using the delete_training_pipeline method.

Java Python

View sample

Delete an endpoint

Deletes an endpoint using the delete_endpoint method.

Java Node.js Python

View sample

Deploy a model

Deploys a model using the deploy_model method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a image classification model
• Get predictions from a image object detection model
• Get predictions from a text classification model
• Get predictions from a text entity extraction model
• Get predictions from a text sentiment analysis model

Deploy a model for custom trained model

Deploys a model for custom trained model using the deploy_model method.

Java Node.js Python

View sample

View in documentation

• Get online predictions and explanations
• Get predictions from a custom trained model
• Get TabNet online predictions
• Get Wide & Deep online predictions

Explain for tabular

Gets explanation for tabular using the explain method.

Python

View sample

Export a model

Exports a model using the export_model method.

Java Node.js Python

View sample

View in documentation

• Export AutoML Edge models

Export a model for tabular classification

Exports a model for tabular classification using the export_model method.

Java Node.js Python

View sample

View in documentation

• Export AutoML tabular models

Export a model for video action recognition

Exports a model for video action recognition using the export_model method.

Java Python

View sample

View in documentation

• Export AutoML Edge models

Get a batch prediction job

Gets a batch prediction job using the get_batch_prediction_job method.

Java Node.js Python

View sample

Get a custom job

Gets a custom job using the get_custom_job method.

Node.js Python

View sample

Get a hyperparameter tuning job

Gets a hyperparameter tuning job using the get_hyperparameter_tuning_job method.

Java Node.js Python

View sample

View in documentation

• Create a hyperparameter tuning job

Get a model

Gets a model using the get_model method.

Java Node.js Python

View sample

View in documentation

• Train an AutoML Edge model using the Vertex AI API

Get a model evaluation

Gets a model evaluation using the get_model_evaluation method.

Java Node.js Python

View sample

Get a model evaluation for image classification

Gets a model evaluation for image classification using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML image classification models

Get a model evaluation for image object detection

Gets a model evaluation for image object detection using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML image object detection models

Get a model evaluation for tabular classification

Gets a model evaluation for tabular classification using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate AutoML classification and regression models

Get a model evaluation for tabular regression

Gets a model evaluation for tabular regression using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate AutoML classification and regression models

Get a model evaluation for text classification

Gets a model evaluation for text classification using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML text classification models

Get a model evaluation for text entity extraction

Gets a model evaluation for text entity extraction using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML text entity extraction models

Get a model evaluation for text sentiment analysis

Gets a model evaluation for text sentiment analysis using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML text sentiment analysis models

Get a model evaluation for video action recognition

Gets a model evaluation for video action recognition using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML video action recognition models

Get a model evaluation for video classification

Gets a model evaluation for video classification using the get_model_evaluation method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML video classification models

Get a model evaluation for video object tracking

Gets a model evaluation for video object tracking using the get_model_evaluation method.

Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML video object tracking models

Get a model evaluation slice

Gets a model evaluation slice using the get_model_evaluation_slice method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML image classification models
• Evaluate and iterate AutoML image object detection models
• Evaluate and iterate AutoML text classification models
• Evaluate and iterate AutoML text entity extraction models
• Evaluate and iterate AutoML text sentiment analysis models
• Evaluate and iterate AutoML video action recognition models
• Evaluate and iterate AutoML video classification models
• Evaluate and iterate AutoML video object tracking models
• Evaluate AutoML classification and regression models

Get a training pipeline

Gets a training pipeline using the get_training_pipeline method.

Java Node.js Python

View sample

View in documentation

• Train an AutoML Edge model using the Vertex AI API

Import data

Imports data using the import_data method.

Node.js Python

View sample

Import data for image classification single label

Imports data for image classification single label using the import_data method.

Python

View sample

Import data for image object detection

Imports data for image object detection using the import_data method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training image object detection models

Import data for text classification single label

Imports data for text classification single label using the import_data method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training text classification models

Import data for text entity extraction

Imports data for text entity extraction using the import_data method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training text entity extraction models

Import data for text sentiment analysis

Imports data for text sentiment analysis using the import_data method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training text sentiment analysis models

Import data for video action recognition

Imports data for video action recognition using the import_data method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training video action recognition models

Import data for video classification

Imports data for video classification using the import_data method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training video classification models

Import data for video object tracking

Imports data for video object tracking using the import_data method.

Java Node.js Python

View sample

View in documentation

• Create a dataset for training video object tracking models

List model evaluation slices

Lists model evaluation slices using the list_model_evaluation_slices method.

Java Node.js Python

View sample

View in documentation

• Evaluate and iterate AutoML image classification models
• Evaluate and iterate AutoML image object detection models
• Evaluate and iterate AutoML text classification models
• Evaluate and iterate AutoML text entity extraction models
• Evaluate and iterate AutoML text sentiment analysis models
• Evaluate and iterate AutoML video action recognition models
• Evaluate and iterate AutoML video classification models
• Evaluate and iterate AutoML video object tracking models
• Evaluate AutoML classification and regression models

Predict

Gets prediction using the predict method.

Python

View sample

Predict for custom trained model

Gets prediction for custom trained model using the predict method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a custom trained model

Predict for image classification

Gets prediction for image classification using the predict method.

Java Node.js Python

View sample

Predict for image object detection

Gets prediction for image object detection using the predict method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a image object detection model

Predict for tabular classification

Gets prediction for tabular classification using the predict method.

Java Node.js Python

View sample

View in documentation

• Get online predictions and explanations
• Get TabNet online predictions
• Get Wide & Deep online predictions

Predict for tabular regression

Gets prediction for tabular regression using the predict method.

Java Node.js Python

View sample

View in documentation

• Get online predictions and explanations
• Get TabNet online predictions
• Get Wide & Deep online predictions

Predict for text classification single label

Gets prediction for text classification single label using the predict method.

Python

View sample

Predict for text entity extraction

Gets prediction for text entity extraction using the predict method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a text entity extraction model

Predict for text sentiment analysis

Gets prediction for text sentiment analysis using the predict method.

Java Node.js Python

View sample

View in documentation

• Get predictions from a text sentiment analysis model

Upload a model

Uploads a model using the upload_model method.

Java Node.js Python

View sample

View in documentation

• Import models to Vertex AI
• Use a custom container for prediction

Upload a model for explain image managed container

Uploads a model for explain image managed container using the upload_model method.

Python

View sample

Upload a model for explain tabular managed container

Uploads a model for explain tabular managed container using the upload_model method.

Python

View sample