diff options
Diffstat (limited to 'src/cascade_creation.py')
| -rw-r--r-- | src/cascade_creation.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/cascade_creation.py b/src/cascade_creation.py index a90939b..1c365b9 100644 --- a/src/cascade_creation.py +++ b/src/cascade_creation.py @@ -28,6 +28,15 @@ class InfluenceGraph(nx.DiGraph): self.add_edges_from(G.edges()) self.add_edges_from([(node_2, node_1) for node_1, node_2 in G.edges()]) + def barabasi_albert(self, n, m): + """ + n : number of nodes + m : number of edges to attach to new node + """ + G = nx.barabasi_albert_graph(n, m) + self.add_edges_from(G.edges()) + self.add_edges_from([(node_2, node_1) for node_1, node_2 in G.edges()]) + def powerlaw_cluster(self, n, m, p, seed=139): """ n : number of nodes |