НачатьНачать бесплатно

Создание системы автодополнения поиска

Автодополнение поиска — распространённое NLP-приложение, которое используется в поисковых системах и мессенджерах. Его цель — предлагать возможные варианты завершения запроса на основе частичного ввода пользователя. Ваша задача — использовать конвейер "text-generation" из Hugging Face для реализации простой системы автодополнения, которая генерирует релевантные варианты завершения по запросу пользователя.

Это упражнение является частью курса

Обработка естественного языка (NLP) на Python

Посмотреть курс

Инструкции к упражнению

  • Создайте конвейер autocomplete с моделью "distilgpt2".
  • Сгенерируйте пять вариантов завершения поискового запроса (suggestions) для заданного prompt, ограничив длину каждого восемью токенами.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

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'])
Редактировать и запускать код