View source on GitHub |
The hyperparameters for an AverageWordEmbeddingClassifier.
Inherits From: BaseHParams
mediapipe_model_maker.text_classifier.AverageWordEmbeddingHParams(
learning_rate: float,
batch_size: int,
epochs: int,
steps_per_epoch: Optional[int] = None,
class_weights: Optional[Mapping[int, float]] = None,
shuffle: bool = False,
repeat: bool = False,
export_dir: str = tempfile.mkdtemp(),
distribution_strategy: str = 'off',
num_gpus: int = 0,
tpu: str = ''
)
Methods
get_strategy
get_strategy()
__eq__
__eq__(
other
)
Class Variables | |
---|---|
class_weights |
None
|
distribution_strategy |
'off'
|
export_dir |
'/tmpfs/tmp/tmpnt_h4p9w'
|
num_gpus |
0
|
repeat |
|