Module: mediapipe_model_maker.text_classifier.model_spec

Specifications for text classifier models.

View aliases

Main aliases

mediapipe_model_maker.python.text.text_classifier.model_spec, mediapipe_model_maker.python.text.text_classifier.text_classifier.ms, mediapipe_model_maker.python.text.text_classifier.text_classifier.text_classifier_options.ms, mediapipe_model_maker.python.text.text_classifier.text_classifier_options.ms, mediapipe_model_maker.text_classifier.text_classifier.ms, mediapipe_model_maker.text_classifier.text_classifier.text_classifier_options.ms, mediapipe_model_maker.text_classifier.text_classifier_options.ms

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