pyagc.metrics.modularity

modularity(edge_index: Tensor, clusters: Tensor, vectorized: bool = True) → float[source]

Computes the modularity of clusters in a graph. This implementation is adapted from: google-research/dmon.

Parameters:

  • edge_index (torch.Tensor) – (2, num_edges), undirected edge list.

  • clusters (torch.Tensor) – (num_nodes,), cluster assignments.

  • vectorized (bool, optional) – Whether to use vectorized computation. (default: True)

Returns:

float – Modularity score.