子集

class dgl.data.utils.Subset(dataset, indices)[source]

Bases: object

在指定索引处数据集的子集

代码改编自PyTorch。

Parameters:

  • dataset – dataset[i] 应该返回第 i 个数据点

  • indices (list) – 用于构建子集的数据点索引列表

__getitem__(item)[source]

获取由项目索引的数据点

Returns:

数据点

Return type:

tuple

__len__()[source]

获取子集大小

Returns:

子集中的数据点数量

Return type:

int