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

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'

Keys to the items in custom_config of Trainer for enabling uCAIP Training.

JOB_ID_KEY 'ai_platform_training_job_id'

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

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'

Keys to the items in custom_config of Trainer for specify the region of uCAIP.