Administra modelos con Vertex AI

Administra modelos de BigQuery ML en el registro de modelos de Vertex AI

Vertex AI es un servicio de Google Cloud que te permite compilar, implementar y escalar modelos de AA. Con Vertex AI, puedes usar herramientas previamente entrenadas y personalizadas, todo dentro de una plataforma unificada. Cuando registras tus modelos de BigQuery ML con Vertex AI Model Registry, puedes administrarlos junto con tus otros modelos de AA para controlar sus versiones, evaluarlos e implementarlos para la predicción con facilidad.

Con esta integración, puedes elegir qué modelos de BigQuery ML registrar en Vertex AI Model Registry. Desde BigQuery ML, puedes registrar lo siguiente:

  • Modelos integrados de BigQuery ML
  • Modelos de TensorFlow de BigQuery ML

Sin embargo, existen algunas limitaciones sobre cuáles se pueden implementar. Las limitaciones de implementación de Vertex AI Model Registry son similares a las limitaciones de exportación de modelos de BigQuery ML. Por ejemplo, los modelos de series temporales de ARIMA no son compatibles naturalmente, así como los modelos de XGboost y otros modelos entrenados con una cláusula de transformación.

Una vez registrado, puedes implementar el modelo de BigQuery ML en un extremo de Vertex AI para la predicción en línea. Para obtener más información sobre la predicción de Vertex AI, consulta la documentación de Vertex AI Prediction.

Para obtener información sobre cómo administrar tus modelos de BigQuery ML desde el registro de modelos de Vertex AI, consulta Introducción a Vertex AI Model Registry.

Requisitos previos de BigQuery ML y Vertex AI Model Registry

Para agregar modelos de BigQuery ML a Vertex AI Model Registry, debes asignar a cloud-dataengine@prod.google.com y cloud-dataengine@system.gserviceaccount.com el rol de administrador de Vertex AI para otorgar los permisos correctos. Esto permite que la cuenta de servicio de BigQuery ML verifique el estado de implementación del modelo. Importante: Esta acción es temporal. En una próxima actualización, podrás usar la credencial de usuario final para registrar BigQuery ML con Vertex AI Model Registry.

Para agregar el permiso, usa este comando de 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'

También deberás habilitar la API de Vertex AI en tu proyecto. Usa este comando de gCloud:


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

Agrega un modelo de BigQuery ML a Vertex AI Model Registry

Para registrar un modelo de BigQuery ML en Vertex AI Model Registry, debes usar model_registry="vertex_ai". Una vez registrado, puedes realizar la implementación desde Vertex AI Model Registry directamente sin importar ni exportar de forma manual. Por ejemplo, puedes ejecutar el siguiente comando para integrar un modelo de BigQuery ML en 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>])

Borra modelos de BigQuery ML de Vertex AI Model Registry

Para borrar un modelo de BigQuery ML de Vertex AI Model Registry, solo necesitas borrarlo de BigQuery ML. Una vez que lo borras de BigQuery ML, se sincroniza y se quita de Vertex AI Model Registry.

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

Consideraciones

¿Qué sucede cuando registro un modelo multirregional de BigQuery ML en Vertex AI Model Registry?

En este momento, si decides agregar un modelo multirregional de BigQuery ML a Vertex AI Model Registry, se convertirá en un modelo regional. Un modelo multirregional de US de BigQuery ML se sincroniza con Vertex AI (us-central1) y un modelo multirregional de EU de BigQuery ML se sincroniza con Vertex AI (europe-west4). En el futuro, Vertex AI Model Registry será compatible con los modelos multirregionales. Para obtener información sobre las ubicaciones compatibles, consulta la página Ubicaciones.

¿Puedo usar las funciones XAI en Vertex AI Model Registry con modelos de BigQuery ML?

No. En este momento, Vertex AI Model Registry no admite XAI con modelos de BigQuery ML.