torch_geometric.datasets.MixHopSyntheticDataset

class MixHopSyntheticDataset(root: str, homophily: float, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

来自“MixHop: Higher-Order Graph Convolutional Architectures via Sparsified Neighborhood Mixing”论文的MixHop合成数据集,包含10个图,每个图具有不同程度的同质性(范围从0.0到0.9)。所有图都有5,000个节点,每个节点对应10个类别中的1个。节点的特征值是从2D高斯分布中采样的,每个类别的特征值都是不同的。

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

  • 同质性 (float) – 同质性的程度(取值为 0.0, 0.1, …, 0.9 之一)。

  • transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before every access. (default: None)

  • pre_transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before being saved to disk. (default: None)

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