class_docstr: Dataset Loader for classification models.
code: def __init__(
      self,
      dataset: tf.data.Dataset,
      label_names: List[str],
      size: Optional[int] = None,
  ):
      super().__init__(dataset, size)
      self._label_names = label_names
  
file_path: mediapipe/mediapipe/model_maker/python/core/data/classification_dataset.py
imports: from typing import List, Optional, Tuple
  import tensorflow
  from mediapipe.model_maker.python.core.data import dataset
label_desc: Use image node_id 1 to create a new ClassificationDataset object from inherited base classes: ds with example: obj = ClassificationDataset(dataset, label_names, size)