forked from mindspore-Ecosystem/mindspore
27 lines
1.4 KiB
ReStructuredText
27 lines
1.4 KiB
ReStructuredText
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 |