tfds.Split

View source on GitHub

Class Split

Enum for dataset splits.

Datasets are typically split into different subsets to be used at various stages of training and evaluation.

  • TRAIN: the training data.
  • VALIDATION: the validation data. If present, this is typically used as evaluation data while iterating on a model (e.g. changing hyperparameters, model architecture, etc.).
  • TEST: the testing data. This is the data to report metrics on. Typically you do not want to use this during model iteration as you may overfit to it.
  • ALL: Special value, never defined by a dataset, but corresponding to all defined splits of a dataset merged together.

See the guide on splits for more information.

__new__

View source

@staticmethod
__new__(
    cls,
    name
)

Create a custom split with tfds.Split('custom_name').

Class Members

  • ALL
  • TEST
  • TRAIN
  • VALIDATION