adding datasets, sorry T-bone

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