mediapipe_model_maker.text_classifier.AverageWordEmbeddingHParams

The hyperparameters for an AverageWordEmbeddingClassifier.

Inherits From: BaseHParams

learning_rate Dataclass field
batch_size Dataclass field
epochs Dataclass field
steps_per_epoch Dataclass field
class_weights Dataclass field
shuffle Dataclass field
repeat Dataclass field
export_dir Dataclass field
distribution_strategy Dataclass field
num_gpus Dataclass field
tpu Dataclass field

Methods

get_strategy

View source

__eq__

class_weights None
distribution_strategy 'off'
export_dir '/tmpfs/tmp/tmpnt_h4p9w'
num_gpus 0
repeat