load_dataset#
- load_dataset(train: bool = True, shuffle: bool = False, object_type: typing_extensions.Literal[VisionData, Dataset] = 'Dataset', n_samples: Optional[int] = None) Union[DatasetV2, VisionData] [source]#
Get the COCO128 dataset and return a dataloader.
- Parameters
- 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, otherwisetf.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.
- Returns
- Union[Dataset, VisionData]
A Dataset or VisionData instance representing COCO128 dataset