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
)
Attributes |
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
)
Class Variables |
hparams
|
None
|
model_options
|
None
|