TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfx.extensions.google_cloud_ai_platform.runner.start_cmle_training

View source on GitHub

Start a trainer job on CMLE.

tfx.extensions.google_cloud_ai_platform.runner.start_cmle_training(
    input_dict,
    output_dict,
    exec_properties,
    executor_class_path,
    training_inputs
)

This is done by forwarding the inputs/outputs/exec_properties to the tfx.scripts.run_executor module on a CMLE training job interpreter.

Args:

  • input_dict: Passthrough input dict for tfx.components.Trainer.executor.
  • output_dict: Passthrough input dict for tfx.components.Trainer.executor.
  • exec_properties: Passthrough input dict for tfx.components.Trainer.executor.
  • executor_class_path: class path for TFX core default trainer.
  • training_inputs: Training input for CMLE training job. 'pythonModule', 'pythonVersion' and 'runtimeVersion' will be inferred by the runner. For the full set of parameters supported, refer to https://cloud.google.com/ml-engine/docs/tensorflow/deploying-models#creating_a_model_version.

Returns:

None

Raises:

  • RuntimeError: if the Google Cloud AI Platform training job failed.