ComenzarEmpieza gratis

Crear un sistema de autocompletado de búsquedas

El autocompletado es una aplicación habitual de NLP en motores de búsqueda y apps de mensajería. Su objetivo es sugerir posibles continuaciones a partir de una entrada parcial del usuario. Tu tarea es usar el pipeline de Hugging Face "text-generation" para implementar un autocompletado básico que genere completions relevantes a partir de la consulta del usuario.

Este ejercicio forma parte del curso

Natural Language Processing (NLP) en Python

Ver curso

Instrucciones del ejercicio

  • Crea un pipeline autocomplete con el modelo "distilgpt2".
  • Genera cinco suggestions de consulta de búsqueda para el prompt dado, limitando cada una a un máximo de ocho tokens.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

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'])
Editar y ejecutar código