Aan de slagGa gratis aan de slag

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

Cursus bekijken

Oefeninstructies

  • Maak een autocomplete-pipeline met het "distilgpt2"-model.
  • Genereer vijf zoekquery-suggestions voor de gegeven prompt, 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'])
Code bewerken en uitvoeren