From 5f1f60a6cbec1140a0fbd90d5fc56dabc2f9888b Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Thu, 5 Nov 2015 16:46:34 -0500 Subject: adding graph level bayes stuff --- simulation/main.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'simulation/main.py') 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. -- cgit v1.2.3-70-g09d2