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

mediapipe_model_maker.text_classifier.AverageWordEmbeddingModelOptions

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

Configurable model options for an Average Word Embedding classifier.

seq_len Length of the sequence to feed into the model.
wordvec_dim Dimension of the word embedding.
do_lower_case Whether to convert all uppercase characters to lowercase during preprocessing.
vocab_size Number of words to generate the vocabulary from data.
dropout_rate The rate for dropout.

Methods

__eq__

do_lower_case True
dropout_rate 0.2
seq_len 256
vocab_size 10000
wordvec_dim 16