Module: tfx.v1.extensions.google_cloud_ai_platform.experimental

Types used in Google Cloud AI Platform under experimental stage.

BULK_INFERRER_SERVING_ARGS_KEY 'ai_platform_serving_args'

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

ENDPOINT_ARGS_KEY 'endpoint'

Keys to the items in custom_config of Pusher/BulkInferrer for optional endpoint override (CAIP).

PUSHER_SERVING_ARGS_KEY 'ai_platform_serving_args'

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

REMOTE_TRIALS_WORKING_DIR_KEY 'remote_trials_working_dir'

Keys to the items in custom_config of Tuner for specifying a working dir for remote trial.

TUNING_ARGS_KEY 'ai_platform_tuning_args'

Keys to the items in custom_config of Tuner 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