mirror of
https://github.com/GoogleCloudPlatform/vertex-ai-samples.git
synced 2026-09-26 14:42:04 +00:00
This sample uses the aiplatform SDK and torch library to implement transformers predictor.
Vertex AI custom prediction routines samples
Overview
Vertex Custom Prediction Routines(CPR) simplify the process of building custom containers and make local model testing easy. Here are the sameple codes for different libraries.
Objectives
The objective is to provide various samples for Vertex Custom Prediction Routine(CPR).
Supporting libraries
- torch
- sklearn
- xgboost