Introdução ao SDK da Vertex AI para Python

O SDK da Vertex AI para Python ajuda a automatizar a ingestão de dados, treinar modelos e receber previsões na Vertex AI. O SDK da Vertex AI usa código Python para acessar a API Vertex AI. Assim, é possível realizar, de maneira programática, a maior parte do que é possível fazer no console do Google Cloud.

Para saber como instalar o SDK da Vertex AI para Python, consulte Instalar o SDK da Vertex AI para Python. Saiba mais na documentação de referência da API SDK da Vertex AI para Python.

Por que usar o SDK Vertex AI

O SDK Vertex AI para Python é recomendado se você for um engenheiro de machine learning (ML) e inteligência artificial (IA) experiente ou um cientista de dados que queira automatizar automaticamente seu fluxo de trabalho. O SDK da Vertex AI para Python é semelhante à biblioteca de cliente Python da Vertex AI, mas o SDK é de nível superior e menos granular. Para ver mais informações, consulte Entenda as diferenças entre o SDK e a biblioteca de cliente.

Escrever código com o SDK da Vertex AI para Python

Para usar o SDK da Vertex AI para Python:

  1. Instale o pacote google-cloud-aiplatform, que inclui o SDK da Vertex AI para Python e a biblioteca de cliente da Vertex AI para Python, executando o seguinte comando no ambiente virtual:

    pip install --upgrade google-cloud-aiplatform
    
  2. Use o código a seguir para importar o namespace 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.

Saiba mais sobre o SDK da Vertex AI para Python

Consulte a seguinte documentação:

Teste as amostras de código e os tutoriais

Os tutoriais do notebook do Jupyter mostram como usar o SDK Vertex AI para Python como parte de um fluxo de trabalho maior. Para mais informações, consulte Tutoriais do notebook da Vertex AI.

As amostras de código no repositório GitHub do SDK Vertex AI para Python mostram como realizar tarefas individuais. Para mais informações, consulte o repositório do SDK do Vertex AI para GitHub.

Entender as diferenças do SDK e da biblioteca de cliente da Vertex AI

Quando você instala o SDK da Vertex AI para Python, a biblioteca de cliente Python da Vertex AI também é instalada. O SDK e a biblioteca de cliente da Vertex AI para Python fornecem funcionalidades semelhantes com diferentes níveis de granularidade. O SDK Vertex AI opera em um nível de abstração mais alto que o da biblioteca de cliente e é adequado para os fluxos de trabalho de ciência de dados mais comuns. Se você precisar de uma funcionalidade de nível inferior, use a biblioteca de cliente Python da Vertex AI.

O SDK Vertex AI está disponível para Python e uma biblioteca de cliente da Vertex AI está disponível para Python, Java e Node.js. Para saber como instalar a biblioteca de cliente Java ou Node.js, consulte Instalar as bibliotecas de cliente da Vertex AI. Se não houver uma biblioteca de cliente disponível na sua linguagem de programação preferida, é possível usar a API REST Vertex AI. Para mais informações, consulte a referência REST da Vertex AI.

Use a biblioteca de cliente Python da Vertex AI e o SDK juntos

Se você usar o SDK Vertex AI para Python e descobrir que precisa de mais flexibilidade ou controle ou se precisar de um método não incluído no SDK Vertex AI, poderá usar a biblioteca de cliente Python da Vertex AI no mesmo fluxo de trabalho. A biblioteca de cliente Python da Vertex AI usa um namespace diferente para acessar a API Vertex AI. A biblioteca de cliente e os namespaces do SDK da Vertex AI para Python podem ser usados no mesmo script em Python, adicionando uma linha import para cada um no seu script Python.

Importar o namespace da biblioteca de cliente Python da Vertex AI

O namespace da biblioteca de cliente Python da Vertex AI é google.cloud.aiplatform.gapic. Esse namespace é mapeado para o namespace google.cloud.aiplatform_v1. Esses dois namespaces podem ser usados de forma intercambiável. Para importar a biblioteca de cliente do Python, inclua uma das seguintes opções no seu script em Python:

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

A seguir