Začněte nyníZačněte zdarma

Vytváření systému pro doplňování vyhledávání

Doplňování vyhledávání, neboli automatické dokončování, je běžná NLP aplikace používaná ve vyhledávačích a chatovacích aplikacích. Cílem je navrhovat možná dokončení na základě částečného vstupu od uživatele. Tvým úkolem je pomocí pipeline "text-generation" z Hugging Face implementovat jednoduchý systém automatického dokončování, který generuje relevantní návrhy na základě zadaného dotazu.

Toto cvičení je součástí kurzu

Zpracování přirozeného jazyka (NLP) v Pythonu

Zobrazit kurz

Pokyny k cvičení

  • Vytvoř pipeline autocomplete s modelem "distilgpt2".
  • Vygeneruj pět návrhů vyhledávacího dotazu suggestions pro zadaný prompt, přičemž každý omezte na maximálně osm tokenů.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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'])
Upravit a spustit kód