是否可以在 Vertex AI Model 注冊表中添加預處理步驟？

Question

我正在嘗試將自定義模型 (XGBoost) 導入 Vertex AI 中的 Model Registry。 我在 BigQuery 中有數據，但在將其傳遞到 model 之前我需要進行一些預處理。
如何導入 model，以便在我部署或根據 BigQuery 中的數據進行批量預測時，它會在進行預測之前進行預處理？

我看到 Model 注冊表將在目標文件夾（例如 .pkl）中搜索具有特定擴展名的 model 文件。 他們還會搜索特定格式的預處理步驟嗎？

我知道 Vertex AI 可以構建管道，所以我需要學習 Kubeflow 來解決嗎？ 比如構建預處理管道並將它們提供給 model。

Answer 1

您可以查看自定義預測例程：

https://cloud.google.com/vertex-ai/docs/predictions/custom-prediction-routines#:~:text=Custom%20prediction%20routines%20(CPR)%20lets,building%20a%20container%20from% 20刮擦。