textcoords="offset points", horizontalalignment="center", verticalalignment="center", xytext=[0, 2], color=cmap_dark(communities[n]), ) plt.axis("off") pathlib.Path("output").mkdir(exist_ok=True) print("Writing network figure to output/karate.png") plt.savefig("output/karate.png") # plt.show() G = nx.karate_club_graph() print("Building Infomap network from a NetworkX graph...") im = Infomap(two_level=True, silent=True, num_trials=20) im.add_networkx_graph(G) print("Find communities with Infomap...") im.run() print( f"Found {im.num_top_modules} modules with codelength {im.codelength:.8f} bits" ) nx.set_node_attributes(G, im.get_modules(), "community") draw_network(G)
import networkx as nx import numpy as np from sklearn.model_selection import ParameterGrid from infomap import Infomap im = Infomap(two_level=True, silent=True, num_trials=10) im.add_networkx_graph(nx.karate_club_graph()) grid = ParameterGrid({"markov_time": np.linspace(0.8, 2, 5)}) for params in grid: im.run(**params) print( f"markov_time={params['markov_time']:0.1f}: number of modules: {im.num_top_modules}" )