mirror of
https://github.com/GoogleCloudPlatform/vertex-ai-samples.git
synced 2026-09-26 14:42:04 +00:00
* feat: PyTorch train and deploy E2E with pre-built containers * address review comments * address review comment 2 * suppress gsutil warning messags * Use unique names * Corrected the project id template
Custom model batch prediction with feature filtering
Learn how to create a custom-trained model from a Python script in a Docker container using the Vertex AI SDK for Python, and then run a batch prediction job by including or excluding a list of features.
The steps performed include:
- Create a Vertex AI custom `TrainingPipeline` for training a model.
- Train a TensorFlow model.
- Send batch prediction job.
Learn more about Vertex AI Batch Prediction.