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

Module: mediapipe_model_maker.text_classifier.model_spec

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

Specifications for text classifier models.

Modules

bert_model_spec module: Specification for a BERT model.

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

mo module: Configurable model options for text classifier models.

Classes

class AverageWordEmbeddingClassifierSpec: Specification for an average word embedding classifier model.

class BertClassifierSpec: Specification for a BERT model.

class SupportedModels: Predefined text classifier model specs supported by Model Maker.

Functions

average_word_embedding_classifier_spec(...): partial(func, *args, **keywords) - new function with partial application

mobilebert_classifier_spec(...): partial(func, *args, **keywords) - new function with partial application