BertNLClassifier

public class BertNLClassifier

L'API Classifier per le attività NLClassification con i modelli Bert, categorizza la stringa in diverse classi. L'API prevede un modello TFLite basato su Bert con metadati popolati.

I metadati devono contenere le seguenti informazioni:

  • 1 input_process_unit per il tokenizzatore di parole/frase.
  • 3 tensori di input con nomi "ids", "mask" e "segment_ids".
  • 1 tensore di output di tipo float32[1, 2], con un file etichetta allegato facoltativamente. Se è allegato un file di etichette, il file deve essere un file di testo normale con un'etichetta per riga, il numero di etichette deve corrispondere al numero di categorie generate dal modello.

Classi nidificate

classe BertNLClassifier.BertNLClassifierOptions Opzioni per configurare BertNLClassifier.

Metodi pubblici

Lista < categoria >
classificare ( String text)
Esegue la classificazione su una stringa di input, restituisce classificato Category s.
static BertNLClassifier
createFromBuffer ( ByteBuffer modelBuffer)
Crea BertNLClassifier con un tampone di modello e predefiniti BertNLClassifier.BertNLClassifierOptions .
static BertNLClassifier
createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions opzioni)
Crea BertNLClassifier con un tampone di modello e BertNLClassifier.BertNLClassifierOptions .
static BertNLClassifier
CreateFromFile (contesto Context, String modelPath)
Crea BertNLClassifier da un file modello con metadati e di default BertNLClassifier.BertNLClassifierOptions .
static BertNLClassifier
CreateFromFile ( file modelFile)
Crea BertNLClassifier da un File oggetto con metadati e di default BertNLClassifier.BertNLClassifierOptions .
static BertNLClassifier
createFromFileAndOptions ( File modelFile, BertNLClassifier.BertNLClassifierOptions opzioni)
Crea BertNLClassifier da un File oggetto con metadati e BertNLClassifier.BertNLClassifierOptions .
static BertNLClassifier
createFromFileAndOptions (contesto Context, String modelPath, BertNLClassifier.BertNLClassifierOptions opzioni)
Crea BertNLClassifier da un file modello con metadati e BertNLClassifier.BertNLClassifierOptions .

Metodi ereditati

Metodi pubblici

pubblico List < categoria > Classifica ( String text)

Esegue la classificazione su una stringa di input, restituisce classificato Category s.

Parametri
testo inserire il testo nel modello.
ritorna
  • Un elenco di risultati di categoria.

public static BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

Crea BertNLClassifier con un tampone di modello e predefiniti BertNLClassifier.BertNLClassifierOptions .

Parametri
modelloBuffer una diretta ByteBuffer o un MappedByteBuffer del modello
ritorna
tiri
IllegalArgumentException se il buffer modello non è una diretta ByteBuffer o un MappedByteBuffer
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public static BertNLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions opzioni)

Crea BertNLClassifier con un tampone di modello e BertNLClassifier.BertNLClassifierOptions .

Parametri
modelloBuffer una diretta ByteBuffer o un MappedByteBuffer del modello
opzioni per configurare il classificatore
ritorna
tiri
IllegalArgumentException se il buffer modello non è una diretta ByteBuffer o un MappedByteBuffer
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public static BertNLClassifier CreateFromFile (contesto Context, String modelPath)

Crea BertNLClassifier da un file modello con metadati e di default BertNLClassifier.BertNLClassifierOptions .

Parametri
contesto Contesto Android
modelloPercorso Percorso al modello di classificazione
ritorna
tiri
IOException Se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public static BertNLClassifier CreateFromFile ( File modelFile)

Crea BertNLClassifier da un File oggetto con metadati e di default BertNLClassifier.BertNLClassifierOptions .

Parametri
modelloFile La classifica modello di File esempio
ritorna
tiri
IOException Se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public static BertNLClassifier createFromFileAndOptions ( File modelFile, BertNLClassifier.BertNLClassifierOptions opzioni)

Crea BertNLClassifier da un File oggetto con metadati e BertNLClassifier.BertNLClassifierOptions .

Parametri
modelloFile La classifica modello di File esempio
opzioni per configurare il classificatore
ritorna
tiri
IOException Se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato

public static BertNLClassifier createFromFileAndOptions (contesto contesto, String modelPath, BertNLClassifier.BertNLClassifierOptions opzioni)

Crea BertNLClassifier da un file modello con metadati e BertNLClassifier.BertNLClassifierOptions .

Parametri
contesto Contesto Android.
modelloPercorso Percorso al modello di classificazione
opzioni per configurare il classificatore
ritorna
tiri
IOException Se il file del modello non viene caricato
IllegalArgumentException se un argomento non è valido
IllegalStateException se c'è un errore interno
RuntimeException se c'è un errore altrimenti non specificato