TextSearcher

lớp cuối cùng công khai TextSearcher

Thực hiện tìm kiếm tương tự trên chuỗi văn bản.

API yêu cầu một mô hình TFLite với Siêu dữ liệu mô hình TFLite tùy chọn nhưng được khuyến khích mạnh mẽ. .

API yêu cầu một mô hình TFLite có siêu dữ liệu được điền sẵn. Siêu dữ liệu phải chứa các thông tin sau:

  • Đối với mô hình TFLite dựa trên Bert:
    • 3 tensor đầu vào loại kTfLiteString có tên "ids", "mask" và "segment_ids".
    • input_process_units cho Trình mã hóa từ/câu
    • chính xác một tensor đầu ra loại kTfLiteFloat32
  • Đối với mô hình TFLite dựa trên Regex:
    • 1 tensor đầu vào.
    • input_process_units cho Trình mã thông báo RegexTokenizer
    • chính xác một tensor đầu ra loại kTfLiteFloat32
  • Đối với mô hình TFLite dựa trên Bộ mã hóa câu phổ quát:
    • 3 tensor đầu vào có tên "inp_text", "res_context" và "res_text"
    • 2 tensor đầu ra có tên "query_encoding" và "response_encoding" loại kTfLiteFloat32

TODO(b/180502532): thêm con trỏ vào mô hình mẫu.

TODO(b/222671076): thêm các phương thức tạo nhà máy không có tùy chọn, chẳng hạn như `createFromFile`, sau khi hỗ trợ định dạng tệp đơn (tệp chỉ mục được đóng gói trong mô hình).

Các lớp lồng nhau

lớp học TextSearcher.TextSearcherOptions Các tùy chọn để thiết lập TextSearcher.

Phương pháp công khai

Trình tìm kiếm văn bản tĩnh
createFromBufferAndOptions (các tùy chọn ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )
Tạo một phiên bản TextSearcher với bộ đệm mô hình và TextSearcher.TextSearcherOptions .
Trình tìm kiếm văn bản tĩnh
Trình tìm kiếm văn bản tĩnh
createFromFileAndOptions (Tùy chọn tệp modelFile, TextSearcher.TextSearcherOptions )
Tạo một phiên bản TextSearcher .
Trình tìm kiếm văn bản tĩnh
createFromFileAndOptions (ngữ cảnh ngữ cảnh, tùy chọn String modelPath, TextSearcher.TextSearcherOptions )
Tạo một phiên bản TextSearcher từ TextSearcher.TextSearcherOptions .
Danh sách < Hàng xóm gần nhất >
tìm kiếm ( Chuỗi văn bản)
Thực hiện trích xuất nhúng trên đầu vào chuỗi được cung cấp, sau đó là tìm kiếm lân cận gần nhất trong chỉ mục.

Phương pháp kế thừa

Phương pháp công khai

TextSearcher tĩnh công khai createFromBufferAndOptions (các tùy chọn ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )

Tạo một phiên bản TextSearcher với bộ đệm mô hình và TextSearcher.TextSearcherOptions .

Thông số
bộ đệm mô hình ByteBuffer trực tiếp hoặc MappedByteBuffer của mô hình tìm kiếm
tùy chọn
Ném
Ngoại lệ Đối số bất hợp pháp nếu bộ đệm mô hình không phải là ByteBuffer trực tiếp hoặc MappedByteBuffer
IOException nếu xảy ra lỗi I/O khi tải tệp chỉ mục
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

TextSearcher tĩnh công khai createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions tùy chọn, int indexFd)

Thông số
bộ đệm mô hình
tùy chọn
chỉ mụcFd

TextSearcher tĩnh công khai createFromFileAndOptions (Tùy chọn tệp modelFile, TextSearcher.TextSearcherOptions )

Tạo một phiên bản TextSearcher .

Thông số
tập tin mô hình mô hình tìm kiếm File tệp
tùy chọn
Ném
IOException nếu xảy ra lỗi I/O khi tải mô hình tflite hoặc tệp chỉ mục
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

TextSearcher tĩnh công khai createFromFileAndOptions (Ngữ cảnh ngữ cảnh, String modelPath, TextSearcher.TextSearcherOptions tùy chọn)

Tạo một phiên bản TextSearcher từ TextSearcher.TextSearcherOptions .

Thông số
bối cảnh
đường dẫn mô hình đường dẫn của mô hình tìm kiếm có siêu dữ liệu trong nội dung
tùy chọn
Ném
IOException nếu xảy ra lỗi I/O khi tải mô hình tflite hoặc tệp chỉ mục
Ngoại lệ Đối số bất hợp pháp nếu một đối số không hợp lệ
Ngoại lệ nhà nước bất hợp pháp nếu có lỗi nội bộ
Ngoại lệ thời gian chạy nếu có một lỗi không xác định khác

Danh sách công khai < Hàng xóm gần nhất > tìm kiếm ( Chuỗi văn bản)

Thực hiện trích xuất nhúng trên đầu vào chuỗi được cung cấp, sau đó là tìm kiếm lân cận gần nhất trong chỉ mục.

Thông số
chữ nhập truy vấn văn bản vào mô hình