load_dataset(train: bool = True, shuffle: bool = False, object_type: Literal['VisionData', 'Dataset'] = 'Dataset', n_samples: Optional[int] = None) Union[DatasetV2, VisionData][source]#

Get the COCO128 dataset and return a dataloader.

trainbool, default: True

if True train dataset, otherwise test dataset

shufflebool, default: False

Whether to shuffle the dataset.

object_typeLiteral[‘Dataset’, ‘Dataset’], default: ‘Dataset’

type of the return value. If ‘Dataset’, deepchecks.vision.VisionData will be returned, otherwise tf.data.Dataset.

n_samplesint, optional

Number of samples to load. Return the first n_samples if shuffle is False otherwise selects n_samples at random. If None, returns all samples.

Union[Dataset, VisionData]

A Dataset or VisionData instance representing COCO128 dataset