BertNLClassifier

klasa publiczna BertNLClassifier

Klasyfikator API dla NLZadania klasyfikacyjne z modelami Berta, kategoryzuje ciągi znaków na różne klasy. Interfejs API oczekuje modelu TFLite opartego na Bercie z wypełnionymi metadanymi.

Metadane powinny zawierać następujące informacje:

  • 1 input_process_unit dla tokenizera Wordpiece/Sentencepiece.
  • 3 tensory wejściowe o nazwach „ids”, „maska” i „segment_ids”.
  • 1 tensor wyjściowy typu float32[1, 2] z opcjonalnie dołączonym plikiem etykiety. Jeśli dołączony jest plik etykiet, powinien to być zwykły plik tekstowy z jedną etykietą w wierszu, a liczba etykiet powinna odpowiadać liczbie kategorii wyprowadzanych przez model.

Klasy zagnieżdżone

klasa BertNLClassifier.BertNLClassifierOptions Opcje konfiguracji BertNLClassifier.

Metody publiczne

Lista < Kategoria >
klasyfikuj (tekst tekstowy )
Wykonuje klasyfikację na wejściu w postaci ciągu znaków, zwraca sklasyfikowaną Category s.
statyczny BertNLClassifier
createFromBuffer ( ByteBuffer modelBuffer)
Tworzy BertNLClassifier z buforem modelu i domyślnym BertNLClassifier.BertNLClassifierOptions .
statyczny BertNLClassifier
createFromBufferAndOptions (opcje ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
Tworzy BertNLClassifier z buforem modelu i BertNLClassifier.BertNLClassifierOptions .
statyczny BertNLClassifier
createFromFile (kontekst kontekstowy, String modelPath)
Tworzy BertNLClassifier z pliku modelu z metadanymi i domyślnym BertNLClassifier.BertNLClassifierOptions .
statyczny BertNLClassifier
createFromFile ( plik modelFile)
Tworzy BertNLClassifier z obiektu File z metadanymi i domyślnym BertNLClassifier.BertNLClassifierOptions .
statyczny BertNLClassifier
createFromFileAndOptions ( Opcje pliku modelFile, BertNLClassifier.BertNLClassifierOptions )
Tworzy BertNLClassifier z obiektu File z metadanymi i BertNLClassifier.BertNLClassifierOptions .
statyczny BertNLClassifier
createFromFileAndOptions (kontekst kontekstowy, string modelPath, opcje BertNLClassifier.BertNLClassifierOptions )
Tworzy BertNLClassifier z pliku modelu z metadanymi i BertNLClassifier.BertNLClassifierOptions .

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 BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

Tworzy BertNLClassifier z buforem modelu i domyślnym BertNLClassifier.BertNLClassifierOptions .

Parametry
Bufor modelu bezpośredni ByteBuffer lub MappedByteBuffer modelu
Zwroty
Rzuca
Wyjątek IllegalArgument jeśli bufor modelu nie jest bezpośrednim ByteBuffer lub MappedByteBuffer
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 BertNLClassifier createFromBufferAndOptions (opcje ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )

Tworzy BertNLClassifier z buforem modelu i BertNLClassifier.BertNLClassifierOptions .

Parametry
Bufor modelu bezpośredni ByteBuffer lub MappedByteBuffer modelu
opcje skonfigurować klasyfikator
Zwroty
Rzuca
Wyjątek IllegalArgument jeśli bufor modelu nie jest bezpośrednim ByteBuffer lub MappedByteBuffer
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 BertNLClassifier createFromFile (kontekst kontekstowy, string modelPath)

Tworzy BertNLClassifier z pliku modelu z metadanymi i domyślnym BertNLClassifier.BertNLClassifierOptions .

Parametry
kontekst Kontekst Androida
Ścieżka modelu Ścieżka do modelu klasyfikacyjnego
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

publiczny statyczny BertNLClassifier createFromFile ( plik modelFile)

Tworzy BertNLClassifier z obiektu File z metadanymi i domyślnym BertNLClassifier.BertNLClassifierOptions .

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 BertNLClassifier createFromFileAndOptions (Opcje pliku modelFile, BertNLClassifier.BertNLClassifierOptions )

Tworzy BertNLClassifier z obiektu File z metadanymi i BertNLClassifier.BertNLClassifierOptions .

Parametry
plik modelu Model klasyfikacji Instancja File
opcje skonfigurować klasyfikator
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 BertNLClassifier createFromFileAndOptions (kontekst kontekstowy, String modelPath, opcje BertNLClassifier.BertNLClassifierOptions )

Tworzy BertNLClassifier z pliku modelu z metadanymi i BertNLClassifier.BertNLClassifierOptions .

Parametry
kontekst Kontekst Androida.
Ścieżka modelu Ścieżka do modelu klasyfikacyjnego
opcje skonfigurować klasyfikator
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