L'API di classificazione per attività di classificazione in linguaggio naturale classifica la stringa in classi diverse.
L'API prevede un modello TFLite con il seguente tensore di input/output:
- Tensore di input (kTfLiteString)
- input del modello, accetta una stringa.
- Tensore del punteggio di output (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
- punteggi di output per ciascuna classe, se type è uno dei tipi Int, dequantizzalo, se è di tipo Bool, converti i valori rispettivamente in 0,0 e 1,0.
- può avere un file associato facoltativo nei metadati per le 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. Tensore etichetta di output: opzionale (kTfLiteString) - nome classe di output per ogni classe, dovrebbe avere la stessa lunghezza dei punteggi. Se questo tensore non è presente, l'API utilizza gli indici di punteggio come nomi di classe. - verrà ignorato se il tensore del punteggio di output ha già un file di etichetta associato.
- il nome della classe di output per ogni classe, dovrebbe avere la stessa lunghezza dei punteggi. Se questo tensore non è presente, l'API utilizza gli indici di punteggio come nomi di classe.
- verrà ignorato se il tensore del punteggio di output ha già un file etichetta associato.
Per impostazione predefinita, l'API tenta di trovare i tensori di input/output con configurazioni predefinite in NLClassifier.NLClassifierOptions
, con il nome del tensore con priorità rispetto all'indice del tensore. L'opzione è configurabile per diversi modelli TFLite.
Classi nidificate
classe | NLClassifier.NLClassifierOptions | Opzioni per identificare i tensori di input e output del modello. |
Metodi pubblici
Elenco <Categoria> | classificare (testo stringa ) Esegue la classificazione su una stringa di input, restituisce le Category classificate. |
Classificatore NL statico | |
Classificatore NL statico | createFromFile (contesto di contesto, stringa modelPath) Crea NLClassifier dal NLClassifier.NLClassifierOptions predefinito. |
Classificatore NL statico | createFromFile ( File modelloFile) Crea NLClassifier dal NLClassifier.NLClassifierOptions predefinito. |
Classificatore NL statico | createFromFileAndOptions (opzioni File modelFile, NLClassifier.NLClassifierOptions ) Crea NLClassifier da NLClassifier.NLClassifierOptions . |
Classificatore NL statico | createFromFileAndOptions (contesto di contesto, stringa modelPath, opzioni NLClassifier.NLClassifierOptions ) Crea NLClassifier da NLClassifier.NLClassifierOptions . |
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
NLClassifier statico pubblico createFromBufferAndOptions ( opzioni ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions )
Crea NLClassifier
con un modello ByteBuffer
e NLClassifier.NLClassifierOptions
.
Parametri
modelBuffer | un ByteBuffer diretto o un MappedByteBuffer del modello di classificazione |
---|---|
opzioni | configurazioni per il modello |
ritorna
- Istanza
NLClassifier
Lancia
IllegalStateException | se c'è un errore interno |
---|---|
RuntimeException | se è presente un errore altrimenti non specificato |
IllegalArgumentException | se il buffer del modello non è un ByteBuffer diretto o un MappedByteBuffer |
public static NLClassifier createFromFile (Contesto di contesto, String modelPath)
Crea NLClassifier
dal NLClassifier.NLClassifierOptions
predefinito.
Parametri
contesto | Contesto Android |
---|---|
modelPath | percorso al modello di classificazione relativo all'asset dir |
ritorna
- un'istanza
NLClassifier
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 NLClassifier createFromFile ( File modelFile)
Crea NLClassifier
dal NLClassifier.NLClassifierOptions
predefinito.
Parametri
modelFile | l'istanza File del modello di classificazione |
---|
ritorna
- un'istanza
NLClassifier
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 NLClassifier createFromFileAndOptions ( file modelFile, opzioni NLClassifier.NLClassifierOptions )
Crea NLClassifier
da NLClassifier.NLClassifierOptions
.
Parametri
modelFile | l'istanza File del modello di classificazione |
---|---|
opzioni | configurazioni per il modello |
ritorna
- un'istanza
NLClassifier
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 NLClassifier createFromFileAndOptions (contesto di contesto, string modelPath, opzioni NLClassifier.NLClassifierOptions )
Crea NLClassifier
da NLClassifier.NLClassifierOptions
.
Parametri
contesto | Contesto Android |
---|---|
modelPath | percorso al modello di classificazione relativo all'asset dir |
opzioni | configurazioni per il modello. |
ritorna
- un'istanza
NLClassifier
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 |