mindspore.dataset.dataloader.RandomSampler

查看源文件
class mindspore.dataset.dataloader.RandomSampler(data_source, replacement=False, num_samples=None, generator=None)[源代码]

对数据集进行随机采样。

参数：

  • data_source (Dataset) - 要从中加载数据的数据集。

  • replacement (bool, 可选) - 是否开启放回采样。默认值： False 。

  • num_samples (Union[int, None], 可选) - 要抽取的样本数量。默认值： None ，将会设置为 data_source 的长度。

  • generator (numpy.random.Generator, 可选) - 采样时使用的生成器。默认值： None 。

样例：

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