Creare un sistema di completamento della ricerca
Il completamento della ricerca, o auto-complete, è un'applicazione NLP comune usata nei motori di ricerca e nelle app di messaggistica. L'obiettivo è suggerire possibili completamenti a partire dall'input parziale dell'utente. Il tuo compito è usare la pipeline "text-generation" di Hugging Face per implementare un sistema di auto-complete di base che generi completamenti pertinenti a partire dalla query dell'utente.
Questo esercizio fa parte del corso
Natural Language Processing (NLP) in Python
Istruzioni dell'esercizio
- Crea una pipeline
autocompletecon il modello"distilgpt2". - Genera cinque
suggestionsdi query di ricerca per ilpromptfornito, limitando ciascuna a un massimo di otto token.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
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'])