Module: tfx.v1.extensions.google_cloud_ai_platform

Google cloud AI platform module.

Modules

experimental module: Types used in Google Cloud AI Platform under experimental stage.

Classes

class BulkInferrer: A Cloud AI component to do batch inference on a remote hosted model.

class Pusher: Component for pushing model to Cloud AI Platform serving.

class Trainer: Cloud AI Platform Trainer component.

class Tuner: TFX component for model hyperparameter tuning on AI Platform Training.

ENABLE_UCAIP_KEY 'ai_platform_training_enable_ucaip'

Deprecated. Please use ENABLE_VERTEX_KEY instead. Keys to the items in custom_config of Trainer for enabling uCAIP Training.

ENABLE_VERTEX_KEY 'ai_platform_enable_vertex'

Keys to the items in custom_config of Trainer and Pusher for enabling Vertex AI.

JOB_ID_KEY 'ai_platform_training_job_id'

Keys to the items in custom_config of Trainer for specifying job id.

LABELS_KEY 'ai_platform_training_labels'

Keys to the items in custom_config of Trainer for specifying labels for training jobs on the AI Platform only. Not applicable for Vertex AI, where labels are specified in the CustomJob as defined in: https://cloud.google.com/vertex-ai/docs/reference/rest/v1/projects.locations.customJobs#CustomJob.

SERVING_ARGS_KEY 'ai_platform_serving_args'

Keys to the items in custom_config of Bulk Inferrer for passing bulkinferrer args to AI Platform.

TRAINING_ARGS_KEY 'ai_platform_training_args'

Keys to the items in custom_config of Trainer for passing training_job to AI Platform, and the GCP project under which the training job will be executed. In Vertex AI, this corresponds to a CustomJob as defined in:https://cloud.google.com/vertex-ai/docs/reference/rest/v1/projects.locations.customJobs#CustomJob.In CAIP, this corresponds to TrainingInputs as defined in:https://cloud.google.com/ml-engine/reference/rest/v1/projects.jobs#TrainingInput

UCAIP_REGION_KEY 'ai_platform_training_ucaip_region'

Deprecated. Please use VERTEX_REGION_KEY instead. Keys to the items in custom_config of Trainer for specifying the region of uCAIP.

VERTEX_CONTAINER_IMAGE_URI_KEY 'ai_platform_vertex_container_image_uri'

Keys to the items in custom_config of Pusher/BulkInferrer for the serving container image URI in Vertex AI.

VERTEX_REGION_KEY 'ai_platform_vertex_region'

Keys to the items in custom_config of Trainer and Pusher for specifying the region of Vertex AI.