Een zoekaanvulsysteem bouwen
Zoekaanvulling, of auto-complete, is een veelvoorkomende NLP-toepassing in zoekmachines en chatapps. Het doel is om mogelijke aanvullingen te suggereren op basis van gedeeltelijke invoer van een gebruiker. Je opdracht is om Hugging Face’s "text-generation"-pipeline te gebruiken om een eenvoudig auto-complete-systeem te implementeren dat relevante aanvullingen genereert op basis van de query van de gebruiker.
Deze oefening maakt deel uit van de cursus
Natural Language Processing (NLP) in Python
Oefeninstructies
- Maak een
autocomplete-pipeline met het"distilgpt2"-model. - Genereer vijf zoekquery-
suggestionsvoor de gegevenprompt, waarbij je elke suggestie beperkt tot maximaal acht tokens.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
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'])