mp.tasks.text.TextClassifierOptions

Options for the text classifier task.

mp.tasks.text.TextClassifierOptions(
    base_options: mp.tasks.BaseOptions,
    display_names_locale: Optional[str] = None,
    max_results: Optional[int] = None,
    score_threshold: Optional[float] = None,
    category_allowlist: Optional[List[str]] = None,
    category_denylist: Optional[List[str]] = None
)

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.

Methods

__eq__

__eq__(
    other
)

Class Variables
category_allowlist	None
category_denylist	None
display_names_locale	None
max_results	None
score_threshold	None