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
- 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
- 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 | createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, tùy chọn TextSearcher.TextSearcherOptions , int indexFd) |
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 > |
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 |
---|