| { |
| "name": "MobileBert text classifier", |
| "description": "Classifies the input string based on the known catergories. To integrate the model into your app, try the `BertNLClassifier` API in the TensorFlow Lite Task library. `BertNLClassifier` takes an input string, and returns the classified label with probability. It encapsulates the processing logic of inputs and outputs and runs the inference with the best practice.", |
| "version": "v1", |
| "subgraph_metadata": [ |
| { |
| "input_tensor_metadata": [ |
| { |
| "name": "ids", |
| "description": "Tokenized ids of input text as concatenated query and passage.", |
| "content": { |
| "content_properties_type": "FeatureProperties", |
| "content_properties": { |
| } |
| } |
| }, |
| { |
| "name": "segment_ids", |
| "description": "0 for query and 1 for passage tokens.", |
| "content": { |
| "content_properties_type": "FeatureProperties", |
| "content_properties": { |
| } |
| } |
| }, |
| { |
| "name": "mask", |
| "description": "Mask with 1 for real tokens and 0 for padding tokens.", |
| "content": { |
| "content_properties_type": "FeatureProperties", |
| "content_properties": { |
| } |
| } |
| } |
| ], |
| "output_tensor_metadata": [ |
| { |
| "name": "probability", |
| "description": "Probabilities of labels respectively.", |
| "content": { |
| "content_properties_type": "FeatureProperties", |
| "content_properties": { |
| } |
| }, |
| "associated_files": [ |
| { |
| "name": "labels.txt", |
| "description": "Labels for classification categories.", |
| "type": "TENSOR_AXIS_LABELS" |
| } |
| ] |
| } |
| ], |
| "input_process_units": [ |
| { |
| "options_type": "BertTokenizerOptions", |
| "options": { |
| "vocab_file": [ |
| { |
| "name": "vocab.txt", |
| "description": "Vocabulary file for the BertTokenizer.", |
| "type": "VOCABULARY" |
| } |
| ] |
| } |
| } |
| ] |
| } |
| ], |
| "author": "TensorFlow", |
| "license": "Apache License. Version 2.0 http://www.apache.org/licenses/LICENSE-2.0.", |
| "min_parser_version": "1.1.0" |
| } |