The Vertex AI SDK also includes classes to create generative AI solutions with text, code, chat, and text embedding foundation models. You can use these classes to generate text, create a text or code chatbot, tune a foundation model, and create a text embedding. A text embedding is text in the form of a vector used to search for items. For more information, see Introduction to language model classes in the Vertex AI SDK.
You can use the Vertex AI SDK for Python in hosted JupyterLab notebooks within Vertex AI to write and run your code. The notebooks include preinstalled ML frameworks, such as TensorFlow and PyTorch. You can also use other notebooks, such as Colab notebooks, or use a developer environment of your choice that supports Python.
If you want to try using the Vertex AI SDK for Python right now, see the following resources:
- Introduction to the Vertex AI SDK for Python
- Vertex AI SDK reference
- Vertex AI SDK language model reference
- Train a model using Vertex AI and the Python SDK
The Vertex AI SDK includes many classes to help you automate data ingestion, train models, and get predictions. It also includes classes to help you monitor, evaluate, and optimize your machine learning (ML) workflow. The classes can be loosely grouped into the following categories:
- Data classes include classes that work with structured data, unstructured data, and the Vertex AI Feature Store.
- Training classes include classes that work with AutoML training for structured and unstructured data, custom training, hyperparameter training, and pipeline training.
- Model classes work with models and model evaluations.
- Prediction classes work with batch predictions, online predictions, and Vector Search predictions.
- Tracking classes work with Vertex ML Metadata, Vertex AI Experiments, and Vertex AI TensorBoard.