子集

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

基类：object

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

代码改编自 PyTorch。

参数:

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

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

__getitem__(item)[source]

获取由 item 索引的数据点

返回:

数据点

返回类型:

tuple

__len__()[source]

获取子集大小

返回:

子集中的数据点数量

返回类型:

int