BertNLClassifier

Bert モデルを使用した NLClassification タスク用の分類子 API は、文字列をさまざまなクラスに分類します。 API は、メタデータが設定された Bert ベースの TFLite モデルを想定しています。

メタデータには次の情報が含まれている必要があります。

• ワードピース/センテンスピーストークナイザー用の 1 input_process_unit。
• 「ids」、「mask」、「segment_ids」という名前を持つ 3 つの入力テンソル。
• float32[1, 2] 型の 1 つの出力テンソル。オプションでラベル ファイルが添付されます。ラベル ファイルが添付される場合、そのファイルは 1 行に 1 つのラベルを含むプレーン テキスト ファイルである必要があり、ラベルの数はモデルが出力するカテゴリの数と一致する必要があります。