python - 如何设置 PipelineJob 的运行名称
问题描述
我有这段代码来启动 VertexAI 管道作业:
import google.cloud.aiplatform as vertexai
vertexai.init(project=PROJECT_ID,staging_bucket=PIPELINE_ROOT)
job = vertexai.PipelineJob(
display_name='pipeline-test-1',
template_path='xgb_pipe.json'
)
job.run()
这很好用,但run name
标签是一个随机数。我该如何指定run name
?
解决方案
name
您可以通过定义管道时的定义来更改“运行名称”中显示的值。
@kfp.dsl.pipeline(name="automl-image-training-v2")
在定义name
using@kfp.dsl.pipeline
时,它会自动附加管道运行的日期和时间。继续编译和运行管道以查看“运行名称”中的更改。
我使用了Vertex AI pipeline examples中的代码。查看管道代码:
@kfp.dsl.pipeline(name="automl-image-training-v2")
def pipeline(project: str = PROJECT_ID, region: str = REGION):
ds_op = gcc_aip.ImageDatasetCreateOp(
project=project,
display_name="flowers",
gcs_source="gs://cloud-samples-data/vision/automl_classification/flowers/all_data_v2.csv",
import_schema_uri=aip.schema.dataset.ioformat.image.single_label_classification,
)
training_job_run_op = gcc_aip.AutoMLImageTrainingJobRunOp(
project=project,
display_name="train-automl-flowers",
prediction_type="classification",
model_type="CLOUD",
base_model=None,
dataset=ds_op.outputs["dataset"],
model_display_name="train-automl-flowers",
training_fraction_split=0.6,
validation_fraction_split=0.2,
test_fraction_split=0.2,
budget_milli_node_hours=8000,
)
endpoint_op = gcc_aip.EndpointCreateOp(
project=project,
location=region,
display_name="train-automl-flowers",
)
gcc_aip.ModelDeployOp(
model=training_job_run_op.outputs["model"],
endpoint=endpoint_op.outputs["endpoint"],
automatic_resources_min_replica_count=1,
automatic_resources_max_replica_count=1,
)
推荐阅读
- amazon-web-services - 如何将 aws ssm 发送命令设置为没有 3600 秒的默认执行时间?
- sql - 与直接在 IN 子句中给出列表的查询相比,雅典娜 IN 子句中的子查询给出不同的结果
- javascript - d3.js 存储从力模拟中随意附加的圆的像素坐标
- git - 重复拉/合并请求后修复存储库
- c++ - 如何解决两个冲突的 QT 版本
- pine-script - 在 PineScript 中使用最高()、最低()的多时间帧变量
- android - 如何限制EditText中点后的数字?
- smb - 除非 vers=1.0,否则无法挂载 cifs smb
- shell - 如何在 csv 中为多个 shell 创建自动编号?
- android - 通过 MIT App Inventor 锁定/解锁 Segway-Ninebot ES2