tfx.extensions.google_cloud_ai_platform.trainer.executor.Executor

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

Inherits From: GenericExecutor, BaseExecutor

tfx.extensions.google_cloud_ai_platform.trainer.executor.Executor(
    context: Optional[tfx.dsl.components.base.base_executor.BaseExecutor.Context] = None
)

Child Classes

class Context

Methods

Do

View source

Do(
    input_dict: Dict[Text, List[types.Artifact]],
    output_dict: Dict[Text, List[types.Artifact]],
    exec_properties: Dict[Text, Any]
)

Starts a trainer job on Google Cloud AI Platform.

Args
input_dict	Passthrough input dict for tfx.components.Trainer.executor.
output_dict	Passthrough input dict for tfx.components.Trainer.executor.
exec_properties	Mostly a passthrough input dict for tfx.components.Trainer.executor. custom_config.ai_platform_training_args and custom_config.ai_platform_training_job_id are consumed by this class. For the full set of parameters supported by Google Cloud AI Platform, refer to https://cloud.google.com/ml-engine/docs/tensorflow/training-jobs#configuring_the_job

Returns
None

Raises
ValueError	if ai_platform_training_args is not in exec_properties.custom_config.
RuntimeError	if the Google Cloud AI Platform training job failed.