torch_geometric.datasets.FakeDataset

class FakeDataset(num_graphs: int = 1, avg_num_nodes: int = 1000, avg_degree: int = 10, 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)[source]

Bases: InMemoryDataset

A fake dataset that returns randomly generated Data objects.

Parameters

  • num_graphs (int, optional) – The number of graphs. (default: 1)

  • avg_num_nodes (int, optional) – The average number of nodes in a graph. (default: 1000)

  • avg_degree (int, optional) – The average degree per node. (default: 10)

  • num_channels (int, optional) – The number of node features. (default: 64)

  • edge_dim (int, optional) – The number of edge features. (default: 0)

  • num_classes (int, optional) – The number of classes in the dataset. (default: 10)

  • task (str, optional) – Whether to return node-level or graph-level labels ("node", "graph", "auto"). If set to "auto", will return graph-level labels if num_graphs > 1, and node-level labels other-wise. (default: "auto")

  • is_undirected (bool, optional) – Whether the graphs to generate are undirected. (default: True)

  • 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)

  • **kwargs (optional) – Additional attributes and their shapes e.g. global_features=5.