forked from huawei/mindspore2022
29 lines
1.1 KiB
ReStructuredText
29 lines
1.1 KiB
ReStructuredText
mindspore.dataset.SubsetSampler
|
||
====================================
|
||
|
||
.. py:class:: mindspore.dataset.SubsetSampler(indices, num_samples=None)
|
||
|
||
给定样本的索引序列，对数据集采样指定索引的样本。
|
||
|
||
**参数：**
|
||
|
||
- **indices** (Iterable): 索引的序列（包括除了string类型的任意Python可迭代对象类型）。
|
||
- **num_samples** (int, 可选): 获取的样本数，可用于部分获取采样得到的样本。默认值：None，获取采样到的所有样本。
|
||
|
||
**异常：**
|
||
|
||
- **TypeError：** `indices` 的类型不是整数。
|
||
- **TypeError：** `num_samples` 不是整数值。
|
||
- **ValueError：** `num_samples` 为负值。
|
||
|
||
**样例：**
|
||
|
||
>>> indices = [0, 1, 2, 3, 4, 5]
|
||
>>>
|
||
>>> # 创建SubsetSampler，根据提供的索引序列，对数据集进行采样
|
||
>>> sampler = ds.SubsetSampler(indices)
|
||
>>> dataset = ds.ImageFolderDataset(image_folder_dataset_dir,
|
||
... num_parallel_workers=8,
|
||
... sampler=sampler)
|
||
|
||
.. include:: mindspore.dataset.BuiltinSampler.rst |