torch_geometric.datasets.BAShapes

class BAShapes(connection_distribution: str = 'random', transform: Optional[Callable] = None)

基础类：BAShapes

BA-Shapes数据集来自“GNNExplainer: Generating Explanations for Graph Neural Networks”论文，包含一个具有300个节点的Barabasi-Albert（BA）图以及一组80个连接到它的“房屋”结构图。

警告

BAShapes 已被弃用，并将在未来的版本中移除。请改用 ExplainerDataset 结合 torch_geometric.datasets.graph_generator.BAGraph。

Parameters:

• connection_distribution (str, 默认: 'random') – 指定房屋和BA图如何连接。有效输入为 "random"（随机选择BA图节点连接到房屋），以及 "uniform"（均匀分布的BA图节点被选择连接到房屋）。

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