Exécuter des exécutions de notebook avec des paramètres

Les instances de notebooks gérés de Vertex AI Workbench vous permettent d'utiliser des valeurs de paramètres dans vos exécutions de notebook pour spécifier les différences d'exécution du code de votre fichier notebook. Cette page explique comment configurer le fichier notebook pour utiliser des paramètres et comment exécuter des exécutions spécifiant des valeurs différentes pour les paramètres de vos notebooks.

Utiliser des paramètres pour exécuter différentes itérations de votre fichier notebook

Vous pouvez utiliser les valeurs des paramètres de notebook dans vos exécutions pour exécuter le même code de notebook, tout en spécifiant des différences telles que :

  • Spécifier un autre ensemble de données à utiliser ou une taille d'échantillon différente.

  • Spécifier différentes configurations de modèle, telles que le taux d'apprentissage ou le type d'optimiseur.

  • Exécuter différents modèles ou différentes versions du même modèle.

Utiliser des paramètres dans une exécution de notebook

Le processus d'exécution d'un notebook avec des paramètres comporte deux étapes principales :

  1. Ajoutez le tag parameters à l'une des cellules de votre fichier notebook. Bien qu'il ne s'agisse pas d'une exigence technique, cette cellule contient généralement du code qui attribue des valeurs à vos variables de paramètre. Si vous n'attribuez pas différentes valeurs de paramètre dans votre exécution, celle-ci utilise les valeurs de paramètre de votre fichier notebook comme valeurs par défaut.

  2. Créez une exécution pour votre fichier notebook incluant de nouvelles valeurs pour vos paramètres. Utilisez le modèle suivant pour mettre en forme vos paramètres et leurs valeurs : parameter1=value1,parameter2=value2. Le format requiert des virgules entre les paires paramètre/valeur, aucun espace et aucun guillemet. Lorsque votre exécution s'exécute, l'exécuteur ajoute une cellule au notebook qui met à jour les valeurs de vos paramètres, directement après la cellule portant le tag parameters.

Avant de commencer

  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. Make sure that billing is enabled for your Google Cloud project.

  4. Enable the Notebooks and Vertex AI APIs.

    Enable the APIs

    5.

  5. In the Google Cloud console, on the project selector page, select or create a Google Cloud 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.
  8. Si vous ne l'avez pas déjà fait, créez une instance de notebooks gérés.

Rôles requis

Pour vous assurer que le compte de service de votre instance dispose des autorisations nécessaires pour interagir avec l'exécuteur Vertex AI Workbench, demandez à votre administrateur d'accorder au compte de service de votre instance le rôle IAM suivant sur le projet :

Pour en savoir plus sur l'attribution de rôles, consultez la page Gérer l'accès aux projets, aux dossiers et aux organisations.

Votre administrateur peut également attribuer au compte de service de votre instance les autorisations requises à l'aide de rôles personnalisés ou d'autres rôles prédéfinis.

Ouvrir JupyterLab

Pour ouvrir JupyterLab et préparer un fichier notebook à exécuter, procédez comme suit.

  1. Ouvrez JupyterLab.

  2. Importez un fichier de notebook (ipynb), ouvrez un fichier existant ou ouvrez un nouveau fichier notebook, puis ajoutez le code que vous souhaitez exécuter sur le nouveau notebook.

  3. Assurez-vous que le code de votre fichier notebook répond aux exigences d'utilisation de l'exécuteur.

Ajouter le tag parameters à une cellule de notebook

  1. Dans l'interface utilisateur de JupyterLab de votre instance de notebooks gérés, ouvrez le fichier de notebook que vous souhaitez exécuter.

  2. Écrivez du code dans une cellule qui attribue des valeurs à vos variables de paramètre. Il s'agit des valeurs utilisées par votre fichier notebook si vous n'attribuez pas différentes valeurs de paramètre lors de votre exécution.

  3. Assurez-vous que la cellule des paramètres est toujours sélectionnée, puis cliquez sur l'Inspecteur de propriétés dans la barre latérale droite.

  4. Dans l'inspection des propriétés, dans la section Tags de cellule, cliquez sur Ajouter un tag, saisissez parameters, puis appuyez sur Enter.

Fournir des valeurs de paramètres pour votre exécution

  1. Dans l'interface utilisateur JupyterLab de votre instance de notebooks gérés, cliquez sur le bouton  Exécuteur.

  2. Dans la boîte de dialogue Envoyer les notebooks à l'exécuteur, saisissez un nom pour votre exécution dans le champ Nom de l'exécution.

  3. Sélectionnez un Type de machine et un Type d'accélérateur.

  4. Sélectionnez un environnement.

  5. Dans le champ Type, sélectionnez Exécution unique ou Exécutions récurrentes basées sur des planifications, puis compléter la boîte de dialogue permettant de planifier l'exécution.

  6. Dans Options avancées, sélectionnez la région dans laquelle vous souhaitez exécuter votre notebook.

  7. Dans le champ Bucket Cloud Storage, sélectionnez un bucket Cloud Storage disponible ou saisissez un nom pour un nouveau bucket, puis cliquez sur Créer et sélectionner. L'exécuteur stocke la sortie de votre notebook dans ce bucket Cloud Storage.

  8. Dans la section Paramètres du notebook, dans la zone de texte Paramètres d'entrée, ajoutez les paramètres des notebooks séparés par une virgule, par exemple, optimizer=SGD,learning_rate=0.01. Le format ne nécessite pas d'espace ni de guillemets.

  9. Configurez le reste de l'exécution, puis cliquez sur Envoyer.

Notebook

Étape suivante