Ejecuta ejecuciones de notebook con parámetros

Ejecuta ejecuciones de notebook con parámetros

Los notebooks administrados de Vertex AI Workbench están obsoletos. El 30 de enero de 2025, finalizará la compatibilidad con notebooks administrados y se quitará la capacidad de crear instancias de notebooks administrados. Las instancias existentes seguirán funcionando, pero los parches y las actualizaciones no estarán disponibles. Para seguir usando Vertex AI Workbench, te recomendamos migrar tus instancias de notebooks administrados a instancias de Vertex AI Workbench.

Las instancias de notebooks administrados de Vertex AI Workbench te permiten usar valores de parámetros en las ejecuciones de notebooks para especificar diferencias en la forma en que se ejecuta el código de tu archivo de notebook. En esta página, se describe cómo configurar tu archivo de notebook a fin de usar parámetros y cómo ejecutar ejecuciones que especifiquen valores diferentes para los parámetros del notebook.

Usa parámetros para ejecutar diferentes iteraciones de tu archivo de notebook

Puedes usar valores de parámetros de notebook en tus ejecuciones para ejecutar el mismo código de notebook y especificar diferencias como las siguientes:

• Especifica un conjunto de datos diferente que quieras usar o un tamaño de muestra diferente del conjunto de datos.

• Especifica diferentes configuraciones de modelos, como la tasa de aprendizaje o el tipo de optimizador.

• Ejecuta diferentes modelos o diferentes versiones del mismo modelo.

Cómo usar parámetros en la ejecución de un notebook

El proceso de ejecución de un notebook con parámetros tiene dos pasos principales:

1. Agrega la etiqueta parameters a una de las celdas de tu archivo de notebook. Si bien no es un requisito técnico, por lo general, esta celda contiene código que asigna valores a las variables de parámetros, aunque este no es un requisito técnico. Si no asignas valores de parámetros diferentes en tu ejecución, la ejecución usa los valores de parámetros en tu archivo de notebook como valores predeterminados.

2. Crea una ejecución para tu archivo de notebook que incluya valores nuevos de tus parámetros. Usa el siguiente patrón para dar formato a los parámetros y sus valores: parameter1=value1,parameter2=value2. El formato requiere comas entre pares de valores de parámetros, sin espacios ni comillas. Cuando se ejecuta la ejecución, el ejecutor agrega una celda al notebook que actualiza los valores de tus parámetros directamente después de la celda etiquetada como parameters.

Antes de comenzar

1. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.

2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

   Note: If you don't plan to keep the resources that you create in this procedure, create a project instead of selecting an existing project. After you finish these steps, you can delete the project, removing all resources associated with the project.

   Go to project selector

3. Make sure that billing is enabled for your Google Cloud project.

4. Enable the Notebooks and Vertex AI APIs.

   Enable the APIs

5. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

   Note: If you don't plan to keep the resources that you create in this procedure, create a project instead of selecting an existing project. After you finish these steps, you can delete the project, removing all resources associated with the project.

   Go to project selector

6. Make sure that billing is enabled for your Google Cloud project.

7. Enable the Notebooks and Vertex AI APIs.

   Enable the APIs

8. Si aún no lo hiciste, crea una instancia de notebooks administrados.

Roles obligatorios

Para garantizar que la cuenta de servicio de tu instancia tenga los permisos necesarios para interactuar con el ejecutor de Vertex AI Workbench, pídele a tu administrador que le otorgue a la cuenta de servicio de tu instancia los siguientes roles de IAM en el proyecto:

• Visualizador de notebooks (roles/notebooks.viewer)
• Usuario de Vertex AI (roles/aiplatform.user)
• Administrador de almacenamiento (roles/storage.admin)

Para obtener más información sobre cómo otorgar roles, consulta Administra el acceso a proyectos, carpetas y organizaciones.

Es posible que tu administrador también pueda otorgar a la cuenta de servicio de tu instancia los permisos necesarios mediante roles personalizados o con otros roles predefinidos.

Abre JupyterLab

Si deseas abrir JupyterLab y preparar un archivo de notebook para que se ejecute, completa los siguientes pasos.

1. Abre JupyterLab.

2. Sube un archivo de notebook (ipynb), abre un archivo existente o abre un archivo de notebook nuevo y agrega al notebook nuevo el código que deseas ejecutar.

3. Asegúrate de que el código del archivo de notebook cumpla con los requisitos para usar el ejecutor.

Agrega la etiqueta parameters a una celda de notebook

1. En la interfaz de usuario de JupyterLab de tu instancia de notebooks administrados, abre el archivo de notebook que deseas ejecutar.

2. Escribe código en una celda que asigne valores a tus variables de parámetros. Estos son los valores que usa el archivo de notebook si no asignas valores de parámetros diferentes en la ejecución.

3. Asegúrate de que la celda de parámetros aún esté seleccionada y, luego, en la barra lateral derecha, haz clic en  Inspector de propiedades.

4. En el inspector de propiedades, en la sección Etiquetas de celda, haz clic en Agregar etiqueta, ingresa parameters y, luego, presiona Enter.

   Nota: Si etiquetas más de una celda con parameters, el ejecutor agrega solo una celda de parámetros, directamente después de la primera celda con parameters.

Proporciona valores de parámetros para tu ejecución

1. En la interfaz de usuario de JupyterLab de tu instancia de notebooks administrados, haz clic en el botón Ejecutor.

2. En el diálogo Enviar notebooks al ejecutor, ingresa un nombre para tu ejecución en el campo Nombre de ejecución.

3. Selecciona un Tipo de máquina y un Tipo de acelerador.

4. Selecciona un entorno.

5. En el campo Tipo, selecciona Ejecución única o Ejecuciones recurrentes basadas en el programa y completa el cuadro de diálogo a fin de programar ejecuciones.

6. En Opciones avanzadas, selecciona la Región en la que deseas ejecutar el notebook.

7. En el campo Bucket de Cloud Storage, selecciona un bucket de Cloud Storage disponible o ingresa un nombre para un bucket nuevo y haz clic en Crear y seleccionar. El ejecutor almacena el resultado de tu notebook en este bucket de Cloud Storage.

8. En la sección Parametrización del notebook, en el cuadro de texto Parámetros de entrada, agrega parámetros de notebook separados por comas, como optimizer=SGD,learning_rate=0.01. El formato requiere que no haya espacios ni comillas.

9. Configura el resto de tu ejecución y, luego, haz clic en Enviar.

Notebook

Para ver un ejemplo de cómo usar los parámetros en una ejecución de notebook como parte de un flujo de trabajo más completo, ejecuta el notebook de Jupyter “Entrena un modelo de clasificación de clases múltiples para la segmentación de anuncios” en uno de los siguientes entornos:

Abrir en Colab |Abrir en Colab Enterprise |Abrir en notebooks administrados por el usuario de Vertex AI Workbench |Ver en GitHub

¿Qué sigue?

• Obtén más información sobre cómo ejecutar el código de notebook en el ejecutor.