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.text_classifier.model_spec.BertClassifierSpec'>, files='https://tfhub.dev/google/mobilebert/uncased_L-24_H-128_B-512_A-4_F-4_OPT/1', hparams=BertHParams(learning_rate=3e-05, batch_size=48, epochs=3, steps_per_epoch=None, class_weights=None, shuffle=False, repeat=False, export_dir='/tmpfs/tmp/tmpnt_h4p9w', distribution_strategy='off', num_gpus=0, tpu='', _strategy=<tensorflow.python.distribute.distribute_lib._DefaultDistributionStrategy object>, end_learning_rate=0.0, optimizer=<BertOptimizer.ADAMW: 'adamw'>, weight_decay=0.01, desired_precisions=[], desired_recalls=[], gamma=2.0, tokenizer=<SupportedBertTokenizers.FULL_TOKENIZER: 'fulltokenizer'>, checkpoint_frequency=0), name='MobileBERT', is_tf2=False)>