Bắt đầu ngayBắt đầu miễn phí

Xây dựng hệ thống gợi ý hoàn thành tìm kiếm

Search completion (tự động hoàn thành) là một ứng dụng NLP phổ biến trong công cụ tìm kiếm và ứng dụng nhắn tin. Mục tiêu là gợi ý các cách hoàn thành khả dĩ dựa trên đầu vào còn dang dở của người dùng. Nhiệm vụ của bạn là dùng pipeline "text-generation" của Hugging Face để triển khai một hệ thống auto-complete cơ bản tạo ra các phần hoàn thành phù hợp từ truy vấn của người dùng.

Bài tập này là một phần của khóa học

Natural Language Processing (NLP) bằng Python

Xem khóa học

Hướng dẫn bài tập

  • Tạo một pipeline autocomplete với mô hình "distilgpt2".
  • Tạo năm suggestions cho truy vấn tìm kiếm từ prompt đã cho, giới hạn mỗi gợi ý tối đa tám token.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

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'])
Chỉnh sửa và Chạy Mã