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 (int, 可选) - 要抽取的样本数量。默认值: None ,将会设置为 data_source 的长度。

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

样例:

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