MediaPipe Public Python API for Text Classifier.
Modules
model_with_tokenizer
module: Text classifier export module library.
Classes
class AverageWordEmbeddingHParams
: The hyperparameters for an AverageWordEmbeddingClassifier.
class AverageWordEmbeddingModelOptions
: Configurable model options for an Average Word Embedding classifier.
class BertHParams
: The hyperparameters for a Bert Classifier.
class BertModelOptions
: Configurable model options for a BERT model.
class BertOptimizer
: Supported Optimizers for Bert Text Classifier.
class CSVParams
: Parameters used when reading a CSV file.
class Dataset
: Dataset library for text classifier.
class SupportedBertTokenizers
: Supported preprocessors.
class SupportedModels
: Predefined text classifier model specs supported by Model Maker.
class TextClassifier
: API for creating and training a text classification model.
class TextClassifierOptions
: User-facing options for creating the text classifier.
