torch_geometric.datasets.FakeDataset

class FakeDataset(num_graphs: int = 1, avg_num_nodes: int = 1000, avg_degree: float = 10.0, num_channels: int = 64, edge_dim: int = 0, num_classes: int = 10, task: str = 'auto', is_undirected: bool = True, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, **kwargs: Union[int, Tuple[int, ...]])[source]

基础类:InMemoryDataset

一个返回随机生成的Data对象的假数据集。

Parameters:
  • num_graphs (int, optional) – 图的数量。(默认值:1

  • avg_num_nodes (int, optional) – 图中节点的平均数量。 (默认值: 1000)

  • avg_degree (float, optional) – 每个节点的平均度数。 (默认值: 10.0)

  • num_channels (int, optional) – 节点特征的数量。 (默认: 64)

  • edge_dim (int, optional) – 边特征的数量。 (默认: 0)

  • num_classes (int, optional) – 数据集中的类别数量。 (默认: 10)

  • 任务 (str, 可选) – 是否返回节点级别或图级别的标签 ("node", "graph", "auto"). 如果设置为 "auto",当 num_graphs > 1 时将返回图级别的标签,否则返回节点级别的标签。 (默认: "auto")

  • is_undirected (bool, optional) – 是否生成无向图。(默认值:True

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

  • **kwargs (可选) – 额外的属性及其形状 例如 global_features=5.