Introducción al SDK de Vertex AI para Python

El SDK de Vertex AI para Python te ayuda a automatizar la ingestión de datos, entrenar modelos y obtener predicciones en Vertex AI. El SDK de Vertex AI usa código de Python para acceder a la API de Vertex AI, de modo que puedas hacer de forma programática la mayoría de las cosas que puedes hacer en la Google Cloud consola.

Para saber cómo instalar o actualizar el SDK de Vertex AI para Python, consulta Instalar el SDK de Vertex AI para Python. Para obtener más información, consulta la documentación de referencia de la API del SDK de Vertex AI para Python.

Por qué usar el SDK de Vertex AI

Se recomienda usar el SDK de Vertex AI para Python si tienes experiencia como ingeniero de aprendizaje automático (ML) e inteligencia artificial (IA) o como científico de datos y quieres automatizar tu flujo de trabajo mediante programación. El SDK de Vertex AI para Python es similar a la biblioteca de cliente de Python de Vertex AI, pero el SDK de Vertex AI es de nivel superior y menos granular. Para obtener más información, consulta Diferencias entre el SDK y la biblioteca de cliente.

Escribir código con el SDK de Vertex AI para Python

Para usar el SDK de Vertex AI para Python, haz lo siguiente:

  1. Instala el paquete google-cloud-aiplatform, que incluye tanto el SDK de Vertex AI para Python como la biblioteca de cliente de Python de Vertex AI, ejecutando el siguiente comando en tu entorno virtual:

    pip install --upgrade google-cloud-aiplatform
    
  2. Usa el siguiente código para importar el espacio de nombres google.cloud.aiplatform:

    from google.cloud import aiplatform
    

  3. If you're using a local shell, then create local authentication credentials for your user account:

    gcloud auth application-default login

    You don't need to do this if you're using Cloud Shell.

    If an authentication error is returned, and you are using an external identity provider (IdP), confirm that you have signed in to the gcloud CLI with your federated identity.

Información sobre el SDK de Vertex AI para Python

Consulta la siguiente documentación:

Probar ejemplos de código y tutoriales

Los tutoriales de cuadernos muestran cómo usar el SDK de Vertex AI para Python como parte de un flujo de trabajo más amplio. Para obtener más información, consulta los tutoriales sobre cuadernos de Vertex AI.

Los ejemplos de código del repositorio de GitHub del SDK de Vertex AI para Python muestran cómo completar tareas concretas. Para obtener más información, consulta el repositorio de GitHub del SDK de Vertex AI para Python.

Conocer las diferencias entre el SDK y la biblioteca de cliente de Vertex AI

Cuando instalas el SDK de Vertex AI para Python, también se instala la biblioteca de cliente de Python de Vertex AI. El SDK de Vertex AI y la biblioteca de cliente de Python de Vertex AI ofrecen funciones similares con diferentes niveles de granularidad. El SDK de Vertex AI opera en un nivel de abstracción superior al de la biblioteca de cliente y es adecuado para la mayoría de los flujos de trabajo de ciencia de datos habituales. Si necesitas funciones de nivel inferior, usa la biblioteca de cliente de Python de Vertex AI.

El SDK de Vertex AI está disponible para Python y hay una biblioteca de cliente de Vertex AI para Python, Java y Node.js. Para saber cómo instalar la biblioteca de cliente de Java o Node.js, consulta el artículo Instalar las bibliotecas de cliente de Vertex AI. Si no hay ninguna biblioteca de cliente disponible en el lenguaje de programación que prefieras, puedes usar la API REST de Vertex AI. Para obtener más información, consulta la referencia de la API REST de Vertex AI.

Usar la biblioteca de cliente de Python y el SDK de Vertex AI juntos

Si usas el SDK de Vertex AI para Python y descubres que necesitas más flexibilidad o control, o si necesitas un método que no está incluido en el SDK de Vertex AI, puedes usar la biblioteca de cliente de Python de Vertex AI en el mismo flujo de trabajo. La biblioteca de cliente de Python de Vertex AI usa un espacio de nombres diferente para acceder a la API Vertex AI. Los espacios de nombres de la biblioteca de cliente y del SDK de Vertex AI para Python se pueden usar en la misma secuencia de comandos de Python añadiendo una línea import para cada uno en la secuencia de comandos de Python.

Importa el espacio de nombres de la biblioteca de cliente de Python de Vertex AI

El espacio de nombres de la biblioteca de cliente de Python de Vertex AI es google.cloud.aiplatform.gapic. Este espacio de nombres se asigna al espacio de nombres google.cloud.aiplatform_v1. Estos dos espacios de nombres se pueden usar indistintamente. Para importar la biblioteca de cliente de Python, incluye una de las siguientes opciones en tu secuencia de comandos de Python:

from google.cloud import aiplatform_v1
from google.cloud.aiplatform import gapic

Siguientes pasos