Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfx.extensions.google_cloud_ai_platform.runner.deploy_model_for_aip_prediction

View source on GitHub

Deploys a model for serving with AI Platform.

tfx.extensions.google_cloud_ai_platform.runner.deploy_model_for_aip_prediction(
    serving_path, model_version, ai_platform_serving_args, executor_class_path
)

Args:

  • serving_path: The path to the model. Must be a GCS URI.
  • model_version: Version of the model being deployed. Must be different from what is currently being served.
  • ai_platform_serving_args: Dictionary containing arguments for pushing to AI Platform. For the full set of parameters supported, refer to https://cloud.google.com/ml-engine/reference/rest/v1/projects.models.versions#Version
  • executor_class_path: class path for TFX core default trainer.

Raises:

  • RuntimeError: if an error is encountered when trying to push.