mindspore.dataset.dataloader.RandomSampler

View Source On Gitee
class mindspore.dataset.dataloader.RandomSampler(data_source, replacement=False, num_samples=None, generator=None)[source]

Samples the dataset elements randomly.

Parameters

  • data_source (Dataset) – Dataset to be sampled.

  • replacement (bool, optional) – Whether to enable the return sampling. Default: False .

  • num_samples (Union[int, None], optional) – Number of samples to be drawn. Default: None , will be set to the length of data_source .

  • generator (mindspore.Generator, optional) – Generator used during sampling. Default: None .

Examples

>>> from mindspore.dataset.dataloader import RandomSampler
>>>
>>> dataset = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> sampler = RandomSampler(dataset)