Bert モデルを使用した NLClassification タスク用の分類子 API は、文字列をさまざまなクラスに分類します。 API は、メタデータが設定された Bert ベースの TFLite モデルを想定しています。
メタデータには次の情報が含まれている必要があります。
- ワードピース/センテンスピーストークナイザー用の 1 input_process_unit。
- 「ids」、「mask」、「segment_ids」という名前を持つ 3 つの入力テンソル。
- float32[1, 2] 型の 1 つの出力テンソル。オプションでラベル ファイルが添付されます。ラベル ファイルが添付される場合、そのファイルは 1 行に 1 つのラベルを含むプレーン テキスト ファイルである必要があり、ラベルの数はモデルが出力するカテゴリの数と一致する必要があります。
ネストされたクラス
クラス | BertNLClassifier.BertNLClassifierOptions | BertNLClassifier を構成するオプション。 |
パブリックメソッド
一覧<カテゴリ> | |
静的BertNLClassifier | createFromBuffer ( ByteBuffer modelBuffer) モデル バッファーとデフォルトの BertNLClassifier.BertNLClassifierOptions を使用してBertNLClassifier を作成します。 |
静的BertNLClassifier | createFromBufferAndOptions ( ByteBuffer modelBuffer、 BertNLClassifier.BertNLClassifierOptionsオプション) モデル バッファーと BertNLClassifier.BertNLClassifierOptions を使用してBertNLClassifier を作成します。 |
静的BertNLClassifier | createFromFile (コンテキスト context、文字列モデルパス) メタデータとデフォルトの BertNLClassifier.BertNLClassifierOptions を含むモデル ファイルからBertNLClassifier を作成します。 |
静的BertNLClassifier | createFromFile (ファイルモデルファイル) |
静的BertNLClassifier | |
静的BertNLClassifier | createFromFileAndOptions (コンテキスト コンテキスト、文字列モデルパス、 BertNLClassifier.BertNLClassifierOptionsオプション) メタデータと BertNLClassifier.BertNLClassifierOptions を含むモデル ファイルからBertNLClassifier を作成します。 |
継承されたメソッド
パブリックメソッド
public static BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)
モデル バッファーとデフォルトのBertNLClassifier.BertNLClassifierOptions
を使用してBertNLClassifier
を作成します。
パラメーター
モデルバッファ | モデルの直接ByteBuffer またはMappedByteBuffer |
---|
戻り値
BertNLClassifier
インスタンス
投げる
IllegalArgumentException | モデル バッファーが直接ByteBuffer またはMappedByteBuffer ではない場合 |
---|---|
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public static BertNLClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer、 BertNLClassifier.BertNLClassifierOptionsオプション)
モデル バッファーとBertNLClassifier.BertNLClassifierOptions
を使用してBertNLClassifier
を作成します。
パラメーター
モデルバッファ | モデルの直接ByteBuffer またはMappedByteBuffer |
---|---|
オプション | 分類器を構成するには |
戻り値
BertNLClassifier
インスタンス
投げる
IllegalArgumentException | モデル バッファーが直接ByteBuffer またはMappedByteBuffer ではない場合 |
---|---|
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public static BertNLClassifier createFromFile (コンテキスト context, String modelPath)
メタデータとデフォルトのBertNLClassifier.BertNLClassifierOptions
を含むモデル ファイルからBertNLClassifier
を作成します。
パラメーター
コンテクスト | Android コンテキスト |
---|---|
モデルパス | 分類モデルへのパス |
戻り値
BertNLClassifier
インスタンス
投げる
IO例外 | モデルファイルの読み込みに失敗した場合 |
---|---|
IllegalArgumentException | 引数が無効な場合 |
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public static BertNLClassifier createFromFile (ファイルモデルファイル)
メタデータとデフォルトのBertNLClassifier.BertNLClassifierOptions
を含むFile
オブジェクトからBertNLClassifier
を作成します。
パラメーター
モデルファイル | 分類モデルFile インスタンス |
---|
戻り値
BertNLClassifier
インスタンス
投げる
IO例外 | モデルファイルの読み込みに失敗した場合 |
---|---|
IllegalArgumentException | 引数が無効な場合 |
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public static BertNLClassifier createFromFileAndOptions (ファイルモデルファイル、 BertNLClassifier.BertNLClassifierOptionsオプション)
メタデータとBertNLClassifier.BertNLClassifierOptions
を含むFile
オブジェクトからBertNLClassifier
を作成します。
パラメーター
モデルファイル | 分類モデルFile インスタンス |
---|---|
オプション | 分類器を構成するには |
戻り値
BertNLClassifier
インスタンス
投げる
IO例外 | モデルファイルの読み込みに失敗した場合 |
---|---|
IllegalArgumentException | 引数が無効な場合 |
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |
public static BertNLClassifier createFromFileAndOptions (コンテキスト コンテキスト、文字列モデルパス、 BertNLClassifier.BertNLClassifierOptionsオプション)
メタデータとBertNLClassifier.BertNLClassifierOptions
を含むモデル ファイルからBertNLClassifier
を作成します。
パラメーター
コンテクスト | Android コンテキスト。 |
---|---|
モデルパス | 分類モデルへのパス |
オプション | 分類器を構成するには |
戻り値
BertNLClassifier
インスタンス
投げる
IO例外 | モデルファイルの読み込みに失敗した場合 |
---|---|
IllegalArgumentException | 引数が無効な場合 |
IllegalStateException | 内部エラーがある場合 |
ランタイム例外 | 他に特定されていないエラーがある場合 |