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. 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.

    Go to project selector

  3. Verify 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.

    Go to project selector

  6. Verify that billing is enabled for your Google Cloud project.

  7. Enable the Notebooks and Vertex AI APIs.

    Enable the APIs

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

    9. 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:

    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.

    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.

    ¿Qué sigue?