mediapipe_model_maker.text_classifier.TextClassifierOptions

User-facing options for creating the text classifier.

mediapipe_model_maker.text_classifier.TextClassifierOptions(
    supported_model: mediapipe_model_maker.text_classifier.SupportedModels,
    hparams: Optional[hp.HParams] = None,
    model_options: Optional[mo.TextClassifierModelOptions] = None
)

supported_model A preconfigured model spec.
hparams Training hyperparameters the user can set to override the ones in supported_model.
model_options Model options the user can set to override the ones in supported_model. The model options type should be consistent with the architecture of the supported_model.

Methods

__eq__

__eq__(
    other
)

hparams None
model_options None