1. Apprendre
    2. /
  2. Cours
    3. /
  3. Travailler avec Hugging Face
Connected

Exercice

Attribution dynamique de catégories

L'attribution dynamique de catégories permet à un modèle de classer un texte dans des catégories prédéfinies, même s'il n'a pas été entraîné au préalable sur ces catégories.

Avec la pipeline() de Hugging Face pour la tâche zero-shot-classification, fournissez le texte et les catégories prédéfinies afin d'identifier la meilleure correspondance.

Créez un classificateur pour prédire l'étiquette du text en entrée, qui est un titre de nouvelle déjà chargé pour vous.

Les pipelines de la bibliothèque transformers sont préchargés pour vous faciliter la tâche.

Remarque : Nous utilisons une version personnalisée de la pipeline pour vous aider à apprendre à utiliser ces fonctions sans avoir à télécharger le modèle.

Instructions

100 XP
  • Construisez la pipeline et enregistrez-la sous classifier.
  • Créez une liste d'étiquettes — "politics", "science", "sports" — et enregistrez-la sous categories.
  • Prédisez l'étiquette de text à l'aide du classificateur et des catégories prédéfinies.