torch_geometric.datasets.graph_generator.TreeGraph

class TreeGraph(depth: int, branch: int = 2, undirected: bool = False)[source]

基础类：GraphGenerator

生成树状图。

Parameters:

  • depth (int) – 树的深度。

  • branch (int, optional) – 树的分支大小。 (默认: 2)

  • 无向 (bool, 可选) – 如果设置为 True，树图将是无向的。（默认值：False）