torch_geometric.datasets.LRGBDataset

class LRGBDataset(root: str, name: str, split: str = 'train', transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, pre_filter: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

“长距离图基准（LRGB）” 数据集是一个包含5个图学习数据集的集合，这些数据集的任务基于图中的长距离依赖关系。有关各个数据集的更多详细信息，请参阅原始 源代码。

数据集

领域

任务

PascalVOC-SP

计算机视觉

节点分类

COCO-SP

计算机视觉

节点分类

PCQM-Contact

量子化学

链接预测

Peptides-func

化学

图分类

Peptides-struct

化学

图回归
Parameters:

  • root (str) – Root directory where the dataset should be saved.

  • name (str) – 数据集的名称（可以是 "PascalVOC-SP", "COCO-SP", "PCQM-Contact", "Peptides-func", "Peptides-struct"）

  • split (str, optional) – 如果为 "train"，则加载训练数据集。 如果为 "val"，则加载验证数据集。 如果为 "test"，则加载测试数据集。 (默认值: "train")

  • transform (callable, optional) – 一个函数/转换，它接收一个 torch_geometric.data.Data 对象并返回一个转换后的 版本。数据对象将在每次访问之前进行转换。 (默认: None)

  • pre_transform (callable, optional) – 一个函数/转换，它接收一个torch_geometric.data.Data对象并返回一个转换后的版本。数据对象将在保存到磁盘之前进行转换。（默认值：None）

  • pre_filter (callable, optional) – 一个函数，接收一个 torch_geometric.data.Data 对象并返回一个布尔值， 表示该数据对象是否应包含在最终数据集中。（默认值：None）

  • force_reload (bool, optional) – Whether to re-process the dataset. (default: False)

统计:

名称

#图表

#节点

#edges

#classes

PascalVOC-SP

11,355

~479.40

~2,710.48

21

COCO-SP

123,286

~476.88

~2,693.67

翻译结果：

~2,693.67

解释：根据翻译规则，公式内容不需要翻译，因此数字和符号保持不变。

81

PCQM-Contact

529,434

~30.14

~61.09

1

Peptides-func

15,535

~150.94

~307.30

10

Peptides-struct

15,535

~150.94

~307.30

11