mediapipe_model_maker.text_classifier.model_spec.AverageWordEmbeddingClassifierSpec

Specification for an average word embedding classifier model.

View aliases

Main aliases

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

mediapipe_model_maker.text_classifier.model_spec.AverageWordEmbeddingClassifierSpec(
    hparams: mediapipe_model_maker.text_classifier.HParams = hp.BaseHParams(epochs=10, batch_size=32, learning_rate=0),
    model_options: mediapipe_model_maker.text_classifier.AverageWordEmbeddingModelOptions = mo.AverageWordEmbeddingModelOptions(),
    name: str = 'AverageWordEmbedding'
)

Attributes
hparams	Configurable hyperparameters for training.
model_options	Configurable options for the average word embedding model.
name	The name of the object.

Methods

__eq__

__eq__(
    other
)

Return self==value.

Class Variables
hparams	Instance of mediapipe_model_maker.text_classifier.HParams
model_options	Instance of mediapipe_model_maker.text_classifier.AverageWordEmbeddingModelOptions
name	'AverageWordEmbedding'