WpfNdZddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZejZejZe jZe j Z!ej"Z#dZ$dZ%dZ&dZ'dZ(ej)GddZ*Gddej+Z,dS)zMediaPipe text classifier task.N)OptionalList)packet_creator) packet_getter)classifications_pb2)classifier_options_pb2)!text_classifier_graph_options_pb2)classification_result base_options) task_info) doc_controls)base_text_task_apiclassifications_outCLASSIFICATIONStext_inTEXTz8mediapipe.tasks.text.text_classifier.TextClassifierGraphceZdZUdZeed<dZeeed<dZ ee ed<dZ ee ed<dZ eeeed<dZeeeed<ejd efd ZdS) TextClassifierOptionsaOptions for the text classifier task. Attributes: base_options: Base options for the text classifier task. display_names_locale: The locale to use for display names specified through the TFLite Model Metadata. max_results: The maximum number of top-scored classification results to return. score_threshold: Overrides the ones provided in the model metadata. Results below this value are rejected. category_allowlist: Allowlist of category names. If non-empty, classification results whose category name is not in this set will be filtered out. Duplicate or unknown category names are ignored. Mutually exclusive with `category_denylist`. category_denylist: Denylist of category names. If non-empty, classification results whose category name is in this set will be filtered out. Duplicate or unknown category names are ignored. Mutually exclusive with `category_allowlist`. r Ndisplay_names_locale max_resultsscore_thresholdcategory_allowlistcategory_denylistreturnc|j}t|j|j|j|j|j}t||S)z3Generates an TextClassifierOptions protobuf object.)rrrrr)r classifier_options) r to_pb2_ClassifierOptionsProtorrrrr _TextClassifierGraphOptionsProto)selfbase_options_protoclassifier_options_protos k/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/mediapipe/tasks/python/text/text_classifier.pyrzTextClassifierOptions.to_pb2Gsk*11336,20!6$  & & & ,'3 5 5 55)__name__ __module__ __qualname____doc__ _BaseOptions__annotations__rrstrrintrfloatrrrrdo_not_generate_docsr rr%r$rr+s&(, ,,,#+x}###%)/8E?))),0htCy)000+/Xd3i(///$ 56 5 5 5%$ 5 5 5r%rcbeZdZdZededdfdZededdfdZdede fdZ d S) TextClassifiera7Class that performs classification on text. This API expects a TFLite model with (optional) TFLite Model Metadata that contains the mandatory (described below) input tensors, output tensor, and the optional (but recommended) category labels as AssociatedFiles with type TENSOR_AXIS_LABELS per output classification tensor. Metadata is required for models with int32 input tensors because it contains the input process unit for the model's Tokenizer. No metadata is required for models with string input tensors. Input tensors: (kTfLiteInt32) - 3 input tensors of size `[batch_size x bert_max_seq_len]` representing the input ids, segment ids, and mask ids - or 1 input tensor of size `[batch_size x max_seq_len]` representing the input ids or (kTfLiteString) - 1 input tensor that is shapeless or has shape [1] containing the input string At least one output tensor with: (kTfLiteFloat32/kBool) - `[1 x N]` array with `N` represents the number of categories. - optional (but recommended) category labels as AssociatedFiles with type TENSOR_AXIS_LABELS, containing one label per line. The first such AssociatedFile (if any) is used to fill the `category_name` field of the results. The `display_name` field is filled from the AssociatedFile (if any) whose locale matches the `display_names_locale` field of the `TextClassifierOptions` used at creation time ("en" by default, i.e. English). If none of these are available, only the `index` field of the results will be filled. model_pathrclt|}t|}||S)aCreates an `TextClassifier` object from a TensorFlow Lite model and the default `TextClassifierOptions`. Args: model_path: Path to the model. Returns: `TextClassifier` object that's created from the model file and the default `TextClassifierOptions`. Raises: ValueError: If failed to create `TextClassifier` object from the provided file such as invalid file path. RuntimeError: If other types of error occurred. )model_asset_pathr )r*rcreate_from_options)clsr3r optionss r$create_from_model_pathz%TextClassifier.create_from_model_pathys8  <<>>G  " "7 + ++r%r8cttdttggdt t gg|}||S)aCreates the `TextClassifier` object from text classifier options. Args: options: Options for the text classifier task. Returns: `TextClassifier` object that's created from `options`. Raises: ValueError: If failed to create `TextClassifier` object from `TextClassifierOptions` such as missing the model. RuntimeError: If other types of error occurred. :) task_graph input_streamsoutput_streams task_options) _TaskInfo_TASK_GRAPH_NAMEjoin _TEXT_TAG_TEXT_IN_STREAM_NAME_CLASSIFICATIONS_TAG_CLASSIFICATIONS_STREAM_NAMEgenerate_graph_config)r7r8r s r$r6z"TextClassifier.create_from_optionssq #xx,@ ABBC HH*,HI J J  I 3y..00 1 11r%textc&|jttj|i}t j}|tj |tt |S)aCPerforms classification on the input `text`. Args: text: The input text. Returns: A `TextClassifierResult` object that contains a list of text classifications. Raises: ValueError: If any of the input arguments is invalid. RuntimeError: If text classification failed to run. ) _runnerprocessrDr create_stringrClassificationResultCopyFromr get_protorFTextClassifierResultcreate_from_pb2)r!rHoutput_packetsclassification_result_protos r$classifyzTextClassifier.classifys\)) ~;DAABDDN#6"J"L"L((/K LMMOOO / /0K L LLr%N) r&r'r(r) classmethodr,r9rr6rPrTr0r%r$r2r2WsB,c,6F,,,;,&2#82=M222;20M3M#7MMMMMMr%r2)-r) dataclassestypingrrmediapipe.pythonrr.mediapipe.tasks.cc.components.containers.protor.mediapipe.tasks.cc.components.processors.protor-mediapipe.tasks.cc.text.text_classifier.protor ,mediapipe.tasks.python.components.containersr classification_result_modulemediapipe.tasks.python.corer base_options_moduler task_info_module1mediapipe.tasks.python.core.optional_dependenciesr mediapipe.tasks.python.text.corerrMrP BaseOptionsr*TextClassifierGraphOptionsr ClassifierOptionsrTaskInfor@rFrErDrCrA dataclassrBaseTextTaskApir2r0r%r$ris&%!!!!!!!!++++++******NNNNNNQQQQQQ[[[[[[nnnnnnKKKKKKEEEEEEJJJJJJ??????3H". #D#_ 0B  % 4(   M (5(5(5(5(5(5(5(5VdMdMdMdMdM'7dMdMdMdMdMr%