forked from huawei/mindspore2022
33 lines
1.5 KiB
ReStructuredText
33 lines
1.5 KiB
ReStructuredText
mindspore.dataset.PKSampler
|
||
==============================
|
||
|
||
.. py:class:: mindspore.dataset.PKSampler(num_val, num_class=None, shuffle=False, class_column='label', num_samples=None)
|
||
|
||
为数据集中的每个P类采样K个元素。
|
||
|
||
**参数：**
|
||
|
||
- **num_val** (int): 每个类要采样的元素数量。
|
||
- **num_class** (int, optional): 要采样的类数量（默认值为None，采样所有类）。当前不支持指定该参数。
|
||
- **shuffle** (bool, optional): 如果为True，则class ID将被打乱，否则它将不会被打乱（默认值为False）。
|
||
- **class_column** (str, optional): 具有MindDataset类标签的列的名称（默认值'label'）。
|
||
- **num_samples** (int, optional): 要采样的样本数（默认值为None，对所有元素进行采样）。
|
||
|
||
**样例：**
|
||
|
||
>>> # 创建一个PKSampler，从每个类中获取3个样本。
|
||
>>> sampler = ds.PKSampler(3)
|
||
>>> dataset = ds.ImageFolderDataset(image_folder_dataset_dir,
|
||
... num_parallel_workers=8,
|
||
... sampler=sampler)
|
||
|
||
**异常：**
|
||
|
||
- **TypeError：** `shuffle` 不是bool值。
|
||
- **TypeError：** `class_column` 不是str值。
|
||
- **TypeError：** `num_samples` 不是整数值。
|
||
- **NotImplementedError：** `num_class` 不为None。
|
||
- **RuntimeError：** `num_val` 不是正值。
|
||
- **ValueError：** `num_samples` 为负值。
|
||
|
||
.. include:: mindspore.dataset.BuiltinSampler.rst |