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
|
Raises |
ValueError
|
if ai_platform_training_args is not in
exec_properties.custom_config.
|
RuntimeError
|
if the Google Cloud AI Platform training job failed.
|