BertNLClassifier

classe pubblica BertNLClassifier

API di classificazione per attività NLClassification con modelli Bert, classifica la stringa in classi diverse. L'API prevede un modello TFLite basato su Bert con metadati popolati.

I metadati dovrebbero contenere le seguenti informazioni:

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

Classi nidificate

classe BertNLClassifier.BertNLClassifierOptions Opzioni per configurare BertNLClassifier.

Metodi pubblici

Elenco <Categoria>
classificare (testo stringa )
Esegue la classificazione su una stringa di input, restituisce le Category classificate.
BertNLClassifier statico
createFromBuffer ( ByteBuffer modelBuffer)
Crea BertNLClassifier con un buffer del modello e BertNLClassifier.BertNLClassifierOptions predefinito.
BertNLClassifier statico
createFromBufferAndOptions ( opzioni ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
Crea BertNLClassifier con un buffer del modello e BertNLClassifier.BertNLClassifierOptions .
BertNLClassifier statico
createFromFile (contesto di contesto, stringa modelPath)
Crea BertNLClassifier da un file modello con metadati e BertNLClassifier.BertNLClassifierOptions .
BertNLClassifier statico
createFromFile ( File modelloFile)
Crea BertNLClassifier da un oggetto File con metadati e BertNLClassifier.BertNLClassifierOptions predefinito.
BertNLClassifier statico
createFromFileAndOptions (opzioni File modelFile, BertNLClassifier.BertNLClassifierOptions )
Crea BertNLClassifier da un oggetto File con metadati e BertNLClassifier.BertNLClassifierOptions .
BertNLClassifier statico
createFromFileAndOptions (contesto contesto, string modelPath, opzioni BertNLClassifier.BertNLClassifierOptions )
Crea BertNLClassifier da un file modello con metadati e BertNLClassifier.BertNLClassifierOptions .

Metodi ereditati

Metodi pubblici

elenco pubblico < categoria > classifica (testo stringa )

Esegue la classificazione su una stringa di input, restituisce le Category classificate.

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

BertNLClassifier statico pubblico createFromBuffer ( ByteBuffer modelBuffer)

Crea BertNLClassifier con un buffer del modello e BertNLClassifier.BertNLClassifierOptions predefinito.

Parametri
modelBuffer un ByteBuffer diretto o un MappedByteBuffer del modello
ritorna
Lancia
IllegalArgumentException se il buffer del modello non è un ByteBuffer diretto o un MappedByteBuffer
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato

BertNLClassifier statico pubblico createFromBufferAndOptions ( opzioni ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )

Crea BertNLClassifier con un buffer del modello e BertNLClassifier.BertNLClassifierOptions .

Parametri
modelBuffer un ByteBuffer diretto o un MappedByteBuffer del modello
opzioni per configurare il classificatore
ritorna
Lancia
IllegalArgumentException se il buffer del modello non è un ByteBuffer diretto o un MappedByteBuffer
IllegalStateException se c'è un errore interno
RuntimeException se è presente un errore altrimenti non specificato

public static BertNLClassifier createFromFile (contesto di contesto, stringa modelPath)

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

Parametri
contesto Contesto Android
modelPath Percorso verso il modello di classificazione
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

BertNLClassifier statico pubblico createFromFile ( File modelFile)

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

Parametri
modelFile Il modello di classificazione Istanza File
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 BertNLClassifier createFromFileAndOptions ( file modelFile, opzioni BertNLClassifier.BertNLClassifierOptions )

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

Parametri
modelFile Il modello di classificazione Istanza File
opzioni per configurare il classificatore
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 BertNLClassifier createFromFileAndOptions (contesto di contesto, stringa modelPath, opzioni BertNLClassifier.BertNLClassifierOptions )

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

Parametri
contesto Contesto Android.
modelPath Percorso verso il modello di classificazione
opzioni per configurare il classificatore
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