scikit_network_louvain_community_detection_karate_club_graph.py

1from IPython.display import SVG
2import numpy as np
3from sknetwork.data import karate_club, visualize_graph
4from sknetwork.clustering import Louvain
5
6# Load the Karate Club graph
7graph = karate_club(metadata=True)
8adjacency = graph.adjacency
9position = graph.position
10
11# Detect communities using Louvain algorithm
12louvain = Louvain()
13labels = louvain.fit_predict(adjacency)
14
15# Visualize the graph with communities
16image = visualize_graph(adjacency, position, labels=labels)
17SVG(image)