torch_geometric.datasets.BAShapes

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

Bases: BAShapes