louvain.py

from collections import defaultdict

from community import best_partition as louvain_partition

from .base import EpochCommunity

# TODO: look into other louvain implementations

class LouvainCommunity(EpochCommunity):
    def __init__(self, epoch=604800, **kwargs):
        super().__init__(epoch, **kwargs)

    def process(self, **kwargs):
        while True:
            yield self.tick()
            g = self.next_epoch(self.env.now)

            partitions = louvain_partition(g, weight='duration')
            communities = defaultdict(set)
            for node, c in partitions.items():
                communities[c].add(node)

            for community in communities.values():
                community = frozenset(community)
                for node in community:
                    self.community[node] = community