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
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
- un'istanza
BertQuestionAnswerer
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
- un'istanza
BertQuestionAnswerer
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)
Crea un'istanza BertQuestionAnswerer
dal BertQuestionAnswerer.BertQuestionAnswererOptions
predefinito.
Parametri
contesto | contesto Android |
---|---|
modelPath | percorso del file al modello con metadati. Nota: il modello non deve essere compresso |
ritorna
- un'istanza
BertQuestionAnswerer
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)
Crea un'istanza BertQuestionAnswerer
dal BertQuestionAnswerer.BertQuestionAnswererOptions
predefinito.
Parametri
modelFile | un oggetto File del modello |
---|
ritorna
- un'istanza
BertQuestionAnswerer
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 )
Crea un'istanza BertQuestionAnswerer
da BertQuestionAnswerer.BertQuestionAnswererOptions
.
Parametri
modelFile | un oggetto File del modello |
---|---|
opzioni |
ritorna
- un'istanza
BertQuestionAnswerer
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 )
Crea un'istanza BertQuestionAnswerer
da BertQuestionAnswerer.BertQuestionAnswererOptions
.
Parametri
contesto | contesto Android |
---|---|
modelPath | percorso del file al modello con metadati. Nota: il modello non deve essere compresso |
opzioni |
ritorna
- un'istanza
BertQuestionAnswerer
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 |