torch_geometric.datasets.Amazon

class Amazon(root: str, name: str, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

来自“图神经网络评估的陷阱”论文的Amazon Computers和Amazon Photo网络。 节点代表商品，边代表两种商品经常一起购买。 给定产品评论作为词袋节点特征，任务是将商品映射到它们各自的产品类别。

Parameters:

  • root (str) – 数据集应保存的根目录。

  • name (str) – 数据集的名称 ("Computers", "Photo").

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

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

  • force_reload (bool, 默认值: False) – 是否重新处理数据集。

统计:

名称

#节点

#edges

#特性

#classes

计算机

13,752

491,722

767

10

照片

7,650

238,162

745

8