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)[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 (float, optional) – The average degree per node. (default:
10.0
)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 ifnum_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
.