Module: tfx.extensions.google_cloud_ai_platform.trainer.executor

View source on GitHub

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.

Other Members

  • JOB_ID_KEY = 'ai_platform_training_job_id'
  • TRAINING_ARGS_KEY = 'ai_platform_training_args'