Subset

class paddle.fluid.dataloader.dataset. 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

Dataset

Example code:

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]