パブリック最終クラスTextSearcher
文字列の類似検索を実行します。
API は、オプションではあるが強く推奨される TFLiteモデル メタデータを備えた TFLite モデルを想定しています。 。
API は、メタデータが設定された TFLite モデルを想定しています。メタデータには次の情報が含まれている必要があります。
- Bert ベースの TFLite モデルの場合:
- 「ids」、「mask」、「segment_ids」という名前を持つ kTfLiteString 型の 3 つの入力テンソル。
- ワードピース/センテンスピーストークナイザーの input_process_units
- kTfLiteFloat32 型の出力テンソルを 1 つだけ
- 1 入力テンソル。
- RegexTokenizer トークナイザーの input_process_units
- kTfLiteFloat32 型の出力テンソルを 1 つだけ
- 「inp_text」、「res_context」、「res_text」という名前の 3 つの入力テンソル
- kTfLiteFloat32 型の名前「query_encoding」および「response_encoding」を持つ 2 つの出力テンソル
TODO(b/180502532): サンプル モデルへのポインターを追加します。
TODO(b/222671076): 単一のファイル形式 (モデルにパックされたインデックス ファイル) がサポートされたら、`createFromFile` などのオプションなしのファクトリ作成メソッドを追加します。
ネストされたクラス
クラス | TextSearcher.TextSearcherOptions | TextSearcher をセットアップするためのオプション。 |
パブリックメソッド
静的TextSearcher | createFromBufferAndOptions ( ByteBuffer modelBuffer、 TextSearcher.TextSearcherOptionsオプション) モデル バッファーと TextSearcher.TextSearcherOptions を使用してTextSearcher インスタンスを作成します。 |
静的TextSearcher | createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer、 TextSearcher.TextSearcherOptionsオプション、intindexFd) |
静的TextSearcher | createFromFileAndOptions (ファイルモデルファイル、 TextSearcher.TextSearcherOptionsオプション) TextSearcher インスタンスを作成します。 |
静的TextSearcher | createFromFileAndOptions (コンテキスト コンテキスト、文字列モデルパス、 TextSearcher.TextSearcherOptionsオプション) TextSearcher.TextSearcherOptions からTextSearcher インスタンスを作成します。 |
リスト<最近隣> |
継承されたメソッド
パブリックメソッド
public static TextSearcher createFromBufferAndOptions ( ByteBuffer modelBuffer、 TextSearcher.TextSearcherOptionsオプション)
モデル バッファーとTextSearcher.TextSearcherOptions
を使用してTextSearcher
インスタンスを作成します。
パラメーター
モデルバッファ | 検索モデルの直接ByteBuffer またはMappedByteBuffer |
---|---|
オプション |
投げる
IllegalArgumentException | モデル バッファーが直接ByteBuffer またはMappedByteBuffer ではない場合 |
---|---|
IO例外 | インデックスファイルのロード時にI/Oエラーが発生した場合 |
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public static TextSearcher createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer、 TextSearcher.TextSearcherOptions options、intindexFd)
パラメーター
モデルバッファ | |
---|---|
オプション | |
インデックスFD |
public static TextSearcher createFromFileAndOptions (ファイルモデルファイル、 TextSearcher.TextSearcherOptionsオプション)
TextSearcher
インスタンスを作成します。
パラメーター
モデルファイル | 検索モデルFile インスタンス |
---|---|
オプション |
投げる
IO例外 | tflite モデルまたはインデックス ファイルのロード時に I/O エラーが発生した場合 |
---|---|
IllegalArgumentException | 引数が無効な場合 |
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public static TextSearcher createFromFileAndOptions (コンテキスト コンテキスト、文字列モデルパス、 TextSearcher.TextSearcherOptionsオプション)
TextSearcher.TextSearcherOptions
からTextSearcher
インスタンスを作成します。
パラメーター
コンテクスト | |
---|---|
モデルパス | アセット内のメタデータを含む検索モデルのパス |
オプション |
投げる
IO例外 | tflite モデルまたはインデックス ファイルのロード時に I/O エラーが発生した場合 |
---|---|
IllegalArgumentException | 引数が無効な場合 |
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public List <NearestNeighbor>検索(文字列テキスト)
指定された文字列入力に対して埋め込み抽出を実行し、続いてインデックス内の最近傍検索を実行します。
パラメーター
文章 | モデルにテキストクエリを入力します |
---|