torch_geometric.datasets.Airports

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

Bases: InMemoryDataset

来自“struc2vec: 从结构身份学习节点表示”论文的机场数据集,其中节点表示机场,标签对应于活动级别。特征由独热编码的节点标识符给出,如“GraLSP: 具有局部结构模式的图神经网络”论文中所述。

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

  • name (str) – 数据集的名称 ("USA", "Brazil", "Europe").

  • transform (Optional[Callable], default: None) – 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.

  • pre_transform (可调用的, 可选的) – 一个函数/转换,它接收 torch_geometric.data.Data 对象并返回一个 转换后的版本。数据对象将在保存到磁盘之前被转换。

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