some minor changes

1 files changed, 11 insertions, 0 deletions

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