CommencerCommencer gratuitement

Créer un système de complétion de recherche

La complétion de recherche, ou l’auto-complétion, est une application courante du NLP utilisée dans les moteurs de recherche et les applications de messagerie. L’objectif est de suggérer des complétions possibles à partir de la saisie partielle d’un utilisateur. Votre tâche consiste à utiliser le pipeline "text-generation" de Hugging Face pour implémenter un système d’auto-complétion basique qui génère des complétions pertinentes à partir de la requête de l’utilisateur.

Cet exercice fait partie du cours

Natural Language Processing (NLP) in Python

Afficher le cours

Instructions

  • Créez un pipeline d’autocomplete avec le modèle "distilgpt2".
  • Générez cinq suggestions de requêtes de recherche pour l’prompt donné, en limitant chacune à un maximum de huit jetons.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

from transformers import pipeline

# Create the pipeline
autocomplete = pipeline(task="____", model="____")

prompt = "Best books to read for"

# Generate search query completions
suggestions = ____

for suggestion in suggestions:
    print(suggestion['generated_text'])
Modifier et exécuter le code