dataeval.utils.dataset#

Provides utility functions for interacting with Computer Vision datasets.

Submodules#

Classes#

SplitDatasetOutput

Output class containing test indices and a list of TrainValSplits

Functions#

read_dataset(dataset)

Extract information from a dataset at each index into individual lists of each information position

split_dataset(labels[, num_folds, stratify, split_on, ...])

Top level splitting function. Returns a dataclass containing a list of train and validation indices.