NLClassifier

klasa publiczna NLClassifier

Klasyfikator API do zadań klasyfikacji języka naturalnego, kategoryzuje ciągi znaków na różne klasy.

API oczekuje modelu TFLite z następującym tensorem wejścia/wyjścia:

  • Tensor wejściowy (kTfLiteString)
    • wejście modelu, akceptuje ciąg znaków.
  • Tensor wyniku wyjściowego (kTfLiteUInt8/kTfLiteInt8/kTfLiteInt16/kTfLiteFloat32/kTfLiteFloat64/kTfLiteBool)
    • wyniki wyjściowe dla każdej klasy, jeśli typ jest jednym z typów Int, zdekwantyzuj go, jeśli jest to typ Bool, przekonwertuj wartości odpowiednio na 0,0 i 1,0.
    • może mieć opcjonalny powiązany plik w metadanych etykiet, plik powinien być zwykłym plikiem tekstowym z jedną etykietą w wierszu, liczba etykiet powinna odpowiadać liczbie kategorii wyprowadzanych przez model. Tensor etykiety wyjściowej: opcjonalny (kTfLiteString) - nazwa klasy wyjściowej dla każdej klasy, powinna mieć tę samą długość z punktacjami. Jeśli tensor nie jest obecny, interfejs API używa indeksów punktacji jako nazw klas. - zostanie zignorowane, jeśli tensor wyniku wyjściowego ma już powiązany plik etykiety.
  • Opcjonalny tensor etykiety wyjściowej (kTfLiteString/kTfLiteInt32)
    • wyjściowa nazwa klasy dla każdej klasy powinna mieć tę samą długość co wyniki. Jeśli tensor nie jest obecny, interfejs API używa indeksów punktacji jako nazw klas.
    • zostanie zignorowany, jeśli tensor wyniku wyjściowego ma już powiązany plik etykiety.

Domyślnie interfejs API próbuje znaleźć tensory wejścia/wyjścia z domyślnymi konfiguracjami w NLClassifier.NLClassifierOptions , z priorytetem nazwy tensora nad indeksem tensora. Opcję można konfigurować dla różnych modeli TFLite.

Klasy zagnieżdżone

klasa NLClassifier.NLClassifierOptions Opcje identyfikacji tensorów wejściowych i wyjściowych modelu.

Metody publiczne

Lista < Kategoria >
klasyfikuj (tekst tekstowy )
Wykonuje klasyfikację na wejściu w postaci ciągu znaków, zwraca sklasyfikowaną Category s.
statyczny NLClassifier
createFromBufferAndOptions (opcje ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions )
Tworzy NLClassifier z modelem ByteBuffer i NLClassifier.NLClassifierOptions .
statyczny NLClassifier
createFromFile (kontekst kontekstowy, String modelPath)
Tworzy NLClassifier z domyślnego NLClassifier.NLClassifierOptions .
statyczny NLClassifier
createFromFile ( plik modelFile)
Tworzy NLClassifier z domyślnego NLClassifier.NLClassifierOptions .
statyczny NLClassifier
createFromFileAndOptions ( Opcje pliku modelFile, NLClassifier.NLClassifierOptions )
Tworzy NLClassifier z NLClassifier.NLClassifierOptions .
statyczny NLClassifier
createFromFileAndOptions (kontekst kontekstowy, String modelPath, opcje NLClassifier.NLClassifierOptions )
Tworzy NLClassifier z NLClassifier.NLClassifierOptions .

Metody dziedziczone

Metody publiczne

lista publiczna <kategoria> klasyfikuj (tekst tekstowy )

Wykonuje klasyfikację na wejściu w postaci ciągu znaków, zwraca sklasyfikowaną Category s.

Parametry
tekst wprowadź tekst do modelu
Zwroty
  • lista wyników kategorii

public static NLClassifier createFromBufferAndOptions (opcje ByteBuffer modelBuffer, NLClassifier.NLClassifierOptions )

Tworzy NLClassifier z modelem ByteBuffer i NLClassifier.NLClassifierOptions .

Parametry
Bufor modelu bezpośredni ByteBuffer lub MappedByteBuffer modelu klasyfikacji
opcje konfiguracje dla danego modelu
Zwroty
Rzuca
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd
Wyjątek IllegalArgument jeśli bufor modelu nie jest bezpośrednim ByteBuffer lub MappedByteBuffer

public static NLClassifier createFromFile (kontekst kontekstowy, string modelPath)

Tworzy NLClassifier z domyślnego NLClassifier.NLClassifierOptions .

Parametry
kontekst Kontekst Androida
Ścieżka modelu ścieżka do modelu klasyfikacji w odniesieniu do katalogu aktywów
Zwroty
Rzuca
Wyjątek IO jeśli plik modelu nie zostanie załadowany
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

public static NLClassifier createFromFile ( plik modelFile)

Tworzy NLClassifier z domyślnego NLClassifier.NLClassifierOptions .

Parametry
plik modelu model klasyfikacji Instancja File
Zwroty
Rzuca
Wyjątek IO jeśli plik modelu nie zostanie załadowany
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

public static NLClassifier createFromFileAndOptions (Opcje pliku modelFile, NLClassifier.NLClassifierOptions )

Tworzy NLClassifier z NLClassifier.NLClassifierOptions .

Parametry
plik modelu model klasyfikacji Instancja File
opcje konfiguracje dla danego modelu
Zwroty
Rzuca
Wyjątek IO jeśli plik modelu nie zostanie załadowany
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd

public static NLClassifier createFromFileAndOptions (kontekst kontekstowy, string modelPath, opcje NLClassifier.NLClassifierOptions )

Tworzy NLClassifier z NLClassifier.NLClassifierOptions .

Parametry
kontekst Kontekst Androida
Ścieżka modelu ścieżka do modelu klasyfikacji w odniesieniu do katalogu aktywów
opcje konfiguracje dla danego modelu.
Zwroty
Rzuca
Wyjątek IO jeśli plik modelu nie zostanie załadowany
Wyjątek IllegalArgument jeśli argument jest nieprawidłowy
Wyjątek IllegalStateException jeśli wystąpił błąd wewnętrzny
Wyjątek czasu wykonania jeśli wystąpi inny nieokreślony błąd