tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package

Example Airflow DAG for Google Vertex AI service testing Custom Jobs operations.

Module Contents

Functions

TABULAR_DATASET(bucket_name)

Attributes

ENV_ID

PROJECT_ID

DAG_ID

REGION

PACKAGE_DISPLAY_NAME

MODEL_DISPLAY_NAME

RESOURCE_DATA_BUCKET

CUSTOM_PYTHON_GCS_BUCKET_NAME

DATA_SAMPLE_GCS_OBJECT_NAME

CONTAINER_URI

MODEL_SERVING_CONTAINER_URI

REPLICA_COUNT

MACHINE_TYPE

ACCELERATOR_TYPE

ACCELERATOR_COUNT

TRAINING_FRACTION_SPLIT

TEST_FRACTION_SPLIT

VALIDATION_FRACTION_SPLIT

PYTHON_PACKAGE_GCS_URI

PYTHON_MODULE_NAME

create_bucket

test_run

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.ENV_ID[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.PROJECT_ID[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.DAG_ID = 'example_vertex_ai_custom_job_operations'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.REGION = 'us-central1'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.PACKAGE_DISPLAY_NAME[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.MODEL_DISPLAY_NAME[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.RESOURCE_DATA_BUCKET = 'airflow-system-tests-resources'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.CUSTOM_PYTHON_GCS_BUCKET_NAME[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.DATA_SAMPLE_GCS_OBJECT_NAME = 'vertex-ai/california_housing_train.csv'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.TABULAR_DATASET(bucket_name)[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.CONTAINER_URI = 'gcr.io/cloud-aiplatform/training/tf-cpu.2-2:latest'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.MODEL_SERVING_CONTAINER_URI = 'gcr.io/cloud-aiplatform/prediction/tf2-cpu.2-2:latest'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.REPLICA_COUNT = 1[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.MACHINE_TYPE = 'n1-standard-4'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.ACCELERATOR_TYPE = 'ACCELERATOR_TYPE_UNSPECIFIED'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.ACCELERATOR_COUNT = 0[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.TRAINING_FRACTION_SPLIT = 0.7[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.TEST_FRACTION_SPLIT = 0.15[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.VALIDATION_FRACTION_SPLIT = 0.15[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.PYTHON_PACKAGE_GCS_URI[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.PYTHON_MODULE_NAME = 'aiplatform_custom_trainer_script.task'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.create_bucket[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job_python_package.test_run[source]

Was this entry helpful?