Ejecuta ejecuciones de notebook con parámetros

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. Accede a tu cuenta de Google Cloud. Si eres nuevo en Google Cloud, crea una cuenta para evaluar el rendimiento de nuestros productos en situaciones reales. Los clientes nuevos también obtienen $300 en créditos gratuitos para ejecutar, probar y, además, implementar cargas de trabajo.

  2. En la página del selector de proyectos de la consola de Google Cloud, selecciona o crea un proyecto de Google Cloud.

    Ir al selector de proyectos

  3. Asegúrate de que la facturación esté habilitada para tu proyecto de Google Cloud.

  4. Habilita las API de Notebooks and Vertex AI.

    Habilita las API

    5.

  5. En la página del selector de proyectos de la consola de Google Cloud, selecciona o crea un proyecto de Google Cloud.

    Ir al selector de proyectos

  6. Asegúrate de que la facturación esté habilitada para tu proyecto de Google Cloud.

  7. Habilita las API de Notebooks and Vertex AI.

    Habilita las API

    8.
  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 el rol de IAM de Visualizador de notebooks () en el proyecto.

Si quieres obtener más información para otorgar roles, consulta Administra el acceso.

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.

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

¿Qué sigue?