From 659f9b4d612dcf5d7e9eed98640be32baa1d108f Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Wed, 4 Feb 2015 10:45:44 -0500 Subject: some minor changes --- src/cascade_creation.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'src/cascade_creation.py') diff --git a/src/cascade_creation.py b/src/cascade_creation.py index df0384e..a90939b 100644 --- a/src/cascade_creation.py +++ b/src/cascade_creation.py @@ -28,6 +28,17 @@ 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 powerlaw_cluster(self, n, m, p, seed=139): + """ + n : number of nodes + m : number of random edges to add for each node + p : proba of adding triangle for each random edge + seed : random number generator + """ + G = nx.powerlaw_cluster_graph(n, m, p, seed) + self.add_edges_from(G.edges()) + self.add_edges_from([(node_2, node_1) for node_1, node_2 in G.edges()]) + def import_from_file(self, file_name): """ Takes a file from the Stanford collection of networks -- cgit v1.2.3-70-g09d2