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

mediapipe_model_maker.text_classifier.SupportedModels

Predefined text classifier model specs supported by Model Maker.

AVERAGE_WORD_EMBEDDING_CLASSIFIER <SupportedModels.AVERAGE_WORD_EMBEDDING_CLASSIFIER: functools.partial(<class 'mediapipe_model_maker.python.text.text_classifier.model_spec.AverageWordEmbeddingClassifierSpec'>)>
MOBILEBERT_CLASSIFIER <SupportedModels.MOBILEBERT_CLASSIFIER: functools.partial(<class 'mediapipe_model_maker.python.text.core.bert_model_spec.BertModelSpec'>, downloaded_files=DownloadedFiles(path='text_classifier/mobilebert_tiny', url='https://storage.googleapis.com/mediapipe-assets/mobilebert_tiny.tar.gz', is_folder=True), hparams=BaseHParams(learning_rate=3e-05, batch_size=48, epochs=3, steps_per_epoch=None, shuffle=False, export_dir='/tmpfs/tmp/tmpqngo3zix', distribution_strategy='off', num_gpus=-1, tpu=''), name='MobileBert', tflite_input_name={'ids': 'serving_default_input_1:0', 'mask': 'serving_default_input_3:0', 'segment_ids': 'serving_default_input_2:0'})>