torch_geometric.datasets.graph_generator.ERGraph

class ERGraph(num_nodes: int, edge_prob: float)[source]

Bases: GraphGenerator

Generates random Erdos-Renyi (ER) graphs. See erdos_renyi_graph() for more information.

Parameters:
  • num_nodes (int) – The number of nodes.

  • edge_prob (float) – Probability of an edge.