开始使用免费开始使用

构建搜索补全系统

搜索补全(也称自动补全)是在搜索引擎和消息应用中常见的 NLP 应用。目标是基于用户的部分输入给出可能的补全。您的任务是使用 Hugging Face 的 "text-generation" pipeline 实现一个基础的自动补全系统,根据用户查询生成相关的补全。

本练习是课程的一部分

Python 中的自然语言处理(NLP)

查看课程

练习说明

  • 使用 "distilgpt2" 模型创建一个 autocomplete pipeline。
  • 针对给定的 prompt 生成 5 条搜索查询 suggestions,并将每条结果限制为最多 8 个 token。

交互式实操练习

通过完成这段示例代码来试试这个练习。

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'])
编辑并运行代码