使用 Vertex AI 管理模型

在 Vertex AI Model Registry 中管理 BigQuery ML 模型

Vertex AI 是一项 Google Cloud 服务,可用于构建、部署和扩缩机器学习模型。借助 Vertex AI,您可以在统一平台中使用预训练和自定义工具。将 BigQuery ML 模型注册到 Vertex AI Model Registry 后,您可以将它们与其他机器学习模型一起管理,以轻松进行版本控制、评估和部署以用于预测。

借助此集成,您可以选择要注册到 Vertex AI Model Registry 的 BigQuery ML 模型。从 BigQuery ML,您可以注册:

  • BigQuery ML 内置模型
  • BigQuery ML TensorFlow 模型

但是,对于可以部署哪些服务存在一些限制。Vertex AI Model Registry 部署限制与 BigQuery ML 模型导出限制类似。例如,ARIMA 时序模型、XGboost 模型以及使用转换子句训练的其他模型均不受支持。

注册后,您可以将 BigQuery ML 模型部署到 Vertex AI 端点以进行在线预测。如需详细了解 Vertex AI 预测,请参阅 Vertex AI 预测文档

如需了解如何从 Vertex AI Model Registry 管理 BigQuery ML 模型,请参阅 Vertex AI Model Registry 简介

BigQuery ML 和 Vertex AI Model Registry 前提条件

要将 BigQuery ML 模型添加到 Vertex AI Model Registry,您必须向 cloud-dataengine@prod.google.com 和 cloud-dataengine@system.gserviceaccount.com 分配 Vertex AI 管理员角色以授予正确的权限。这样,BigQuery ML 服务帐号才能查看模型部署状态。重要提示:这是临时性措施。在未来版本中,您将能够使用最终用户凭据向 Vertex AI Model Registry 注册 BigQuery ML。

如需添加权限,请使用以下 gCloud 命令:

gcloud projects add-iam-policy-binding PROJECT_NUMBER \
      --member='serviceAccount:cloud-dataengine@system.gserviceaccount.com' \
      --role='roles/aiplatform.admin'

gcloud projects add-iam-policy-binding PROJECT_NUMBER \
      --member='user:cloud-dataengine@prod.google.com' \
      --role='roles/aiplatform.admin'

您还需要在项目中启用 Vertex AI API。请使用此 gCloud 命令:


gcloud --project PROJECT_ID services enable  aiplatform.googleapis.com

将 BigQuery ML 模型添加到 Vertex AI Model Registry

如需将 BigQuery ML 模型注册到 Vertex AI Model Registry,您必须使用 model_registry="verex_ai"。注册后,您可以直接从 Vertex AI Model Registry 部署,而无需手动导出或导入。例如,您可以运行以下命令将 BigQuery ML 模型与 Vertex AI Model Registry 集成:

   CREATE OR REPLACE MODEL [PROJECT_ID].[DATASET_ID].[BQML_MODEL_ID]
   OPTIONS(model_registry="vertex_ai",
   [vertex_ai_model_id=<vertex_ai_model_id>,
   vertex_ai_model_version_aliases=[<version_aliases>])

从 Vertex AI Model Registry 中删除 BigQuery ML 模型

要从 Vertex AI Model Registry 中删除 BigQuery ML 模型,您只需从 BigQuery ML 中删除该模型。从 BigQuery ML 中删除模型后,它会同步到 Vertex AI Model Registry 并从中移除。

   DROP MODEL [PROJECT_ID].[DATASET_ID].[BQML_MODEL_ID]

注意事项

当我将多区域 BigQuery ML 模型注册到 Vertex AI Model Registry 时会发生什么?

目前,如果您决定将多区域 BigQuery ML 模型添加到 Vertex AI Model Registry,它会将该模型转换为单区域模型。BigQuery ML 多区域美国模型同步到 Vertex AI (us-central1),BigQuery ML 多区域欧盟模型同步到 Vertex AI (europe-west4)。Vertex AI Model Registry 将来会支持多区域模型。如需了解受支持的位置,请参阅位置页面。

我可以将 Vertex AI Model Registry 中的 XAI 功能与 BigQuery ML 模型一起使用吗?

否,目前 Vertex AI Model Registry 不支持将 XAI 与 BigQuery ML 模型搭配使用。