TextSearcher

TextSearcher kelas akhir publik

Melakukan pencarian kesamaan pada string teks.

API mengharapkan model TFLite dengan Metadata Model TFLite opsional, namun sangat disarankan. .

API mengharapkan model TFLite dengan metadata yang terisi. Metadata harus berisi informasi berikut:

  • Untuk model TFLite berbasis Bert:
    • 3 input tensor bertipe kTfLiteString dengan nama "ids", "mask" dan "segment_ids".
    • input_process_units untuk Tokenizer Bagian Kata/Kalimat
    • tepat satu tensor keluaran tipe kTfLiteFloat32
  • Untuk model TFLite berbasis Regex:
    • 1 tensor masukan.
    • input_process_units untuk Tokenizer RegexTokenizer
    • tepat satu tensor keluaran tipe kTfLiteFloat32
  • Untuk model TFLite berbasis Universal Sentence Encoder:
    • 3 tensor masukan dengan nama "inp_text", "res_context" dan "res_text"
    • 2 tensor keluaran dengan nama "query_encoding" dan "response_encoding" bertipe kTfLiteFloat32

TODO(b/180502532): menambahkan penunjuk ke model contoh.

TODO(b/222671076): menambahkan metode pembuatan pabrik tanpa opsi, seperti `createFromFile`, setelah format file tunggal (file indeks yang dikemas dalam model) didukung.

Kelas Bersarang

kelas TextSearcher.TextSearcherOptions Opsi untuk menyiapkan TextSearcher.

Metode Publik

Pencari Teks statis
Pencari Teks statis
Pencari Teks statis
Pencari Teks statis
Daftar < Tetangga Terdekat >
pencarian ( String teks)
Melakukan ekstraksi penyematan pada input string yang disediakan, diikuti dengan pencarian tetangga terdekat di indeks.

Metode Warisan

Metode Publik

TextSearcher statis publik createFromBufferAndOptions ( opsi ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )

Membuat instance TextSearcher dengan buffer model dan TextSearcher.TextSearcherOptions .

Parameter
modelBuffer ByteBuffer langsung atau MappedByteBuffer dari model pencarian
pilihan
Melempar
Pengecualian Argumen Ilegal jika buffer model bukan ByteBuffer langsung atau MappedByteBuffer
Pengecualian IO jika terjadi kesalahan I/O saat memuat file indeks
IlegalStateException jika ada kesalahan internal
Pengecualian Waktu Proses jika ada kesalahan yang tidak ditentukan

TextSearcher statis publik createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opsi TextSearcher.TextSearcherOptions , int indexFd)

Parameter
modelBuffer
pilihan
indeksFd

TextSearcher statis publik createFromFileAndOptions ( File modelFile, opsi TextSearcher.TextSearcherOptions )

Membuat instance TextSearcher .

Parameter
file model contoh File model pencarian
pilihan
Melempar
Pengecualian IO jika terjadi kesalahan I/O saat memuat model tflite atau file indeks
Pengecualian Argumen Ilegal jika suatu argumen tidak valid
IlegalStateException jika ada kesalahan internal
Pengecualian Waktu Proses jika ada kesalahan yang tidak ditentukan

public static TextSearcher createFromFileAndOptions (Konteks konteks, String modelPath, opsi TextSearcher.TextSearcherOptions )

Parameter
konteks
modelPath jalur model pencarian dengan metadata di aset
pilihan
Melempar
Pengecualian IO jika terjadi kesalahan I/O saat memuat model tflite atau file indeks
Pengecualian Argumen Ilegal jika suatu argumen tidak valid
IlegalStateException jika ada kesalahan internal
Pengecualian Waktu Proses jika ada kesalahan yang tidak ditentukan

Daftar publik <NearestNeighbor> pencarian ( String teks)

Melakukan ekstraksi penyematan pada input string yang disediakan, diikuti dengan pencarian tetangga terdekat di indeks.

Parameter
teks memasukkan kueri teks ke model