Bygga ett system för sökförslag
Sökförslag, eller autokomplettering, är ett vanligt NLP-tillämpningsområde som används i sökmotorer och meddelandeappar. Målet är att föreslå möjliga kompletteringar utifrån användarens delvis inmatade text. Din uppgift är att använda Hugging Faces "text-generation"-pipeline för att implementera ett enkelt autokompletteringssystem som genererar relevanta förslag baserade på användarens sökfråga.
Den här övningen är en del av kursen
Natural Language Processing (NLP) i Python
Övningsinstruktioner
- Skapa en
autocomplete-pipeline med modellen"distilgpt2". - Generera fem sökförslags
suggestionsför det angivnaprompt, och begränsa varje förslag till maximalt åtta token.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
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'])