ComenzarEmpieza gratis

Creación de un sistema de autocompletado de búsquedas

La finalización de la búsqueda, o autocompletar, es una aplicación común del PLN utilizada en motores de búsqueda y aplicaciones de mensajería. El objetivo es sugerir posibles completaciones basadas en la entrada parcial de un usuario. Tu tarea consiste en utilizar el proceso de Hugging Face « "text-generation" » para implementar un sistema básico de autocompletado que genere completamientos relevantes a partir de la consulta del usuario.

Este ejercicio forma parte del curso

Procesamiento del lenguaje natural (NLP) en Python

Ver curso

Instrucciones del ejercicio

  • Crea una canalización de autocomplete con el modelo "distilgpt2".
  • Genera cinco consultas de búsqueda suggestions para el término dado prompt, 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