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

Module: mediapipe_model_maker.text_classifier.text_classifier

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

API for text classification.

Modules

classifier module: Custom classifier.

ds module: Common dataset for model training and evaluation.

hp module: Hyperparameters for training models. Shared across tasks.

mo module: Configurable model options for text classifier models.

model_util module: Utilities for models.

ms module: Specifications for text classifier models.

preprocessor module: Preprocessors for text classification.

quantization module: Libraries for post-training quantization.

text_classifier_options module: User-facing customization options to create and train a text classifier.

text_ds module: Text classifier dataset library.

Classes

class TextClassifier: API for creating and training a text classification model.