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")transform (callable, optional) – 一个函数/转换,它接收一个
torch_geometric.data.Data对象并返回一个转换后的版本。数据对象将在每次访问之前进行转换。(默认值:None)**kwargs (可选) – 额外的属性及其形状 例如
global_features=5.