Subset¶

class paddle.io. Subset ( dataset, indices ) [source]

Subset of a dataset at specified indices.

Parameters

• dataset (Dataset) – The whole Dataset.

• indices (sequence) – Indices in the whole set selected for subset.

Returns

A Dataset which is the subset of the original dataset.

Return type

List[Dataset]

Examples

import paddle
from paddle.io import Subset

# example 1:
a = paddle.io.Subset(dataset=range(1, 4), indices=[0, 2])
print(list(a))
# [1, 3]

# example 2:
b = paddle.io.Subset(dataset=range(1, 4), indices=[1, 1])
print(list(b))
# [2, 2]