BertQuestionAnswerer

classe pubblica BertQuestionAnswerer

Restituisce il maggior numero di risposte possibili a una determinata domanda per i modelli QA (BERT, Albert, ecc.).

L'API prevede un modello TFLite basato su Bert con metadati contenenti le seguenti informazioni:

  • input_process_units per Wordpiece/Sentencepiece Tokenizer - Wordpiece Tokenizer può essere utilizzato per un modello MobileBert , Sentencepiece Tokenizer Tokenizer può essere utilizzato per un modello Albert .
  • 3 tensori di input con nomi "ids", "mask" e "segment_ids".
  • 2 tensori di output con nomi "end_logits" e "start_logits".

Classi nidificate

classe BertQuestionAnswerer.BertQuestionAnswererOptions Opzioni per impostare un BertQuestionAnswerer .

Metodi pubblici

Elenco <QaAnswer>
risposta (contesto stringa , domanda stringa )
Risponde alla domanda in base al contesto e restituisce un elenco di possibili QaAnswer .
BertQuestionAnswerer statico
createAlbertQuestionAnswererFromFile (contesto di contesto, stringa modelPath, stringa frasePieceModelPath)
Crea un'istanza BertQuestionAnswerer con un modello Albert e un file modello della parte della frase.
BertQuestionAnswerer statico
createBertQuestionAnswererFromFile (Contesto contesto, String modelPath, String vocabPath)
Crea un'istanza BertQuestionAnswerer con un modello Bert e un file di vocabolario.
BertQuestionAnswerer statico
createFromFile (contesto di contesto, stringa modelPath)
BertQuestionAnswerer statico
BertQuestionAnswerer statico
BertQuestionAnswerer statico

Metodi ereditati

Metodi pubblici

elenco pubblico <QaAnswer> risposta (contesto stringa , domanda stringa )

Risponde alla domanda in base al contesto e restituisce un elenco di possibili QaAnswer . Potrebbe essere vuoto se non è stata trovata alcuna risposta dal contesto fornito.

Parametri
contesto contesto su cui si basa la domanda
domanda domanda da porre
ritorna
  • un elenco di possibili risposte in QaAnswer

public static BertQuestionAnswerer createAlbertQuestionAnswererFromFile (contesto di contesto, stringa modelPath, stringa frasePieceModelPath)

Crea un'istanza BertQuestionAnswerer con un modello Albert e un file modello della parte della frase.

Un modello adatto è: https://tfhub.dev/tensorflow/lite-model/albert_lite_base/squadv1/1

Parametri
contesto contesto Android
modelPath percorso del file al modello Albert. Nota: il modello non deve essere compresso
frasePieceModelPath percorso del file del modello della parte della frase. Nota: il modello non deve essere compresso
ritorna
Lancia
IOException Se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato

public static BertQuestionAnswerer createBertQuestionAnswererFromFile (contesto di contesto, stringa modelPath, stringa vocabPath)

Crea un'istanza BertQuestionAnswerer con un modello Bert e un file di vocabolario.

Un modello adatto è: https://tfhub.dev/tensorflow/lite-model/mobilebert/1/default/1

Parametri
contesto contesto Android
modelPath percorso del file al modello Bert. Nota: il modello non deve essere compresso
vocabPath percorso del file del vocabolario. Nota: il file non deve essere compresso
ritorna
Lancia
IOException Se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato

public static BertQuestionAnswerer createFromFile (Contesto di contesto, String modelPath)

Parametri
contesto contesto Android
modelPath percorso del file al modello con metadati. Nota: il modello non deve essere compresso
ritorna
Lancia
IOException se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato

BertQuestionAnswerer pubblico statico createFromFile ( File modelFile)

Parametri
modelFile un oggetto File del modello
ritorna
Lancia
IOException se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato

public static BertQuestionAnswerer createFromFileAndOptions ( file modelFile, opzioni BertQuestionAnswerer.BertQuestionAnswererOptions )

Parametri
modelFile un oggetto File del modello
opzioni
ritorna
Lancia
IOException se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato

public static BertQuestionAnswerer createFromFileAndOptions (contesto di contesto, string modelPath, opzioni BertQuestionAnswerer.BertQuestionAnswererOptions )

Parametri
contesto contesto Android
modelPath percorso del file al modello con metadati. Nota: il modello non deve essere compresso
opzioni
ritorna
Lancia
IOException se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato