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

Module: tfx.extensions.google_cloud_ai_platform.trainer.executor

Helper class to start TFX training jobs on AI Platform.

Classes

class Executor: Start a trainer job on Google Cloud AI Platform using a default Trainer.

class GenericExecutor: Start a trainer job on Google Cloud AI Platform using a generic Trainer.

ENABLE_UCAIP_KEY 'ai_platform_training_enable_ucaip'
JOB_ID_KEY 'ai_platform_training_job_id'
TRAINING_ARGS_KEY 'ai_platform_training_args'
UCAIP_REGION_KEY 'ai_platform_training_ucaip_region'