CommencerCommencer gratuitement

Développement d'un système de complétion de recherche

La complétion de recherche, ou saisie semi-automatique, est une application courante du TALN utilisée dans les moteurs de recherche et les applications de messagerie. L'objectif est de proposer des compléments possibles en fonction de la saisie partielle d'un utilisateur. Votre tâche consiste à utiliser le pipeline d'"text-generation" de Hugging Face pour mettre en œuvre un système d'auto-complétion de base qui génère des complétions pertinentes à partir de la requête de l'utilisateur.

Cet exercice fait partie du cours

Traitement du langage naturel (NLP) en Python

Afficher le cours

Instructions

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

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