Aan de slagBegin gratis

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

Bekijk cursus

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.

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan 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