Join the SIG TFX-Addons community and help make TFX even better!

Module: tfx.extensions.google_cloud_ai_platform.runner

Helper class to start TFX training jobs on AI Platform.


create_model_for_aip_prediction_if_not_exist(...): Creates a new model for serving with AI Platform if not exists.

delete_model_from_aip_if_exists(...): Deletes a model from Google Cloud AI Platform if exists.

delete_model_version_from_aip_if_exists(...): Deletes a model version from Google Cloud AI Platform if version exists.

deploy_model_for_aip_prediction(...): Deploys a model for serving with AI Platform.

get_service_name_and_api_version(...): Gets service name and api version from ai_platform_serving_args.

start_aip_training(...): Start a trainer job on AI Platform (AIP).