IniziaInizia gratis

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

Visualizza il corso

Istruzioni dell'esercizio

  • Crea una pipeline autocomplete con il modello "distilgpt2".
  • Genera cinque suggestions di query di ricerca per il prompt fornito, 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'])
Modifica ed esegui il codice