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
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
- un'istanza
BertNLClassifier
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
- un'istanza
BertNLClassifier
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
- un'istanza
BertNLClassifier
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
- un'istanza
BertNLClassifier
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
- un'istanza
BertNLClassifier
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
- un'istanza
BertNLClassifier
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 |