如何將參數傳遞到 Vertex AI Platform Pipeline？

Question

我創建了一個與此類似的 Vertex AI 管道。

現在管道引用了 csv 文件。 因此，如果此 csv 文件發生更改，則需要重新創建管道。

有什么方法可以在重新運行時將新的 csv 作為參數傳遞給管道？ 那是沒有使用筆記本重新創建管道？

如果沒有，是否有自動更新數據集、model 和部署的最佳實踐方法？

Answer 1

看看那個文檔。

你可以這樣定義你的管道

...
# Define the workflow of the pipeline.
@kfp.dsl.pipeline(
    name="automl-image-training-v2",
    pipeline_root=pipeline_root_path)
def pipeline(project_id: str):
...

（你的筆記本樣本中有一些非常相似的東西）

然后，當您調用管道時，您可以傳遞一些參數

import google.cloud.aiplatform as aip

job = aip.PipelineJob(
    display_name="automl-image-training-v2",
    template_path="image_classif_pipeline.json",
    pipeline_root=pipeline_root_path,
    parameter_values={
        'project_id': project_id
    }
)

job.submit()

您可以在參數值和管道 function 的參數中看到project_id一個 dict 參數。

對您的 CSV 文件名執行相同操作！