构建搜索补全系统
搜索补全(也称自动补全)是在搜索引擎和消息应用中常见的 NLP 应用。目标是基于用户的部分输入给出可能的补全。您的任务是使用 Hugging Face 的 "text-generation" pipeline 实现一个基础的自动补全系统,根据用户查询生成相关的补全。
本练习是课程的一部分
Python 中的自然语言处理(NLP)
练习说明
- 使用
"distilgpt2"模型创建一个autocompletepipeline。 - 针对给定的
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'])