Join the SIG TFX-Addons community and help make TFX even better! Join SIG TFX-Addons

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.

SERVING_ARGS_KEY 'ai_platform_serving_args'

Keys to the items in custom_config of Pusher/BulkInferrer for passing serving args to AI Platform.

TRAINING_ARGS_KEY 'ai_platform_training_args'

Keys to the items in custom_config of Trainer for passing training_args to AI Platform.

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.