CommencezCommencez gratuitement

Créer un système de saisie semi-automatique pour la recherche

La saisie semi-automatique (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 proposer 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 créer un système d'auto-complétion de base qui génère des suggestions pertinentes à partir de la requête de l'utilisateur.

Cette activité fait partie du cours

Natural Language Processing (NLP) en Python

Voir le cours

Instructions de l’exercice

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

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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