scikit_network_karate_club_louvain_community_detection_visualization.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
10labels_true = graph.labels
11
12# Community detection using Louvain algorithm
13louvain = Louvain()
14labels = louvain.fit_predict(adjacency)
15
16# Visualization
17image = visualize_graph(adjacency, position, labels=labels)
18SVG(image)