Attention: This MediaPipe Solutions Preview is an early release. Learn more.

mp.tasks.text.TextClassifierOptions

Stay organized with collections Save and categorize content based on your preferences.

Options for the text classifier task.

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.

Methods

__eq__

category_allowlist None
category_denylist None
display_names_locale None
max_results None
score_threshold None