From a819813ca6c310d84a52de51e7bc49ea8dd8a726 Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Wed, 4 Feb 2015 12:35:03 -0500 Subject: adding datasets, sorry T-bone --- src/cascade_creation.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/cascade_creation.py') 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 -- cgit v1.2.3-70-g09d2