adding graph level bayes stuff

1 files changed, 9 insertions, 0 deletions

diff --git a/simulation/main.py b/simulation/main.py
index 7aa107f..2425cf6 100644
--- a/simulation/main.py
+++ b/simulation/main.py
@@ -77,6 +77,15 @@ def build_matrix(cascades, node):
     return x, y
 
 
+def build_cascade_list(cascades):
+    x, s = [], []
+    for cascade in cascades:
+        xlist, slist = zip(*cascade)
+        x.append(xlist)
+        s.append(slist)
+    return x, s
+
+
 def simulate_cascade(x, graph):
     """
     Simulate an IC cascade given a graph and initial state.