Subset

class dgl.data.utils.Subset(dataset, indices)

Bases: object

Subset of a dataset at specified indices

Code adapted from PyTorch.

Parameters:

• dataset – dataset[i] should return the ith datapoint

• indices (list) – List of datapoint indices to construct the subset

__getitem__(item)

Get the datapoint indexed by item

Returns:

datapoint

Return type:

tuple

__len__()

Get subset size

Returns:

Number of datapoints in the subset

Return type:

int