View source on GitHub |
A generic dataset class for loading model training and evaluation dataset.
mediapipe_model_maker.model_util.dataset.Dataset(
tf_dataset: tf.data.Dataset, size: Optional[int] = None
)
For each ML task, such as image classification, text classification etc., a subclass can be derived from this class to provide task-specific data loading utilities.
Attributes | |
---|---|
size
|
Returns the size of the dataset.
Same functionality as calling len. See the len method definition for more information. |
Methods
gen_tf_dataset
gen_tf_dataset(
batch_size: int = 1,
is_training