Final commit before submission

1 files changed, 20 insertions, 8 deletions

diff --git a/simulation/main.py b/simulation/main.py
index bdc5c97..e4b30f2 100644
--- a/simulation/main.py
+++ b/simulation/main.py
@@ -125,22 +125,34 @@ if __name__ == "__main__":
     g = np.array([[0, 0, 1], [0, 0, 0.5], [0, 0, 0]])
     p = 0.5
     g = np.log(1. / (1 - p * g))
-    sizes = [100, 500, 1000, 5000, 10000]
     # error = []
 
-    def source(graph):
+    def source(graph, t):
         x0 = np.zeros(graph.shape[0], dtype=bool)
         a = randint(0, 1)
         x0[a] = True
-        if random() > 0.01:
+        if random() > t:
             x0[1-a] = True
         return x0
 
-    for i in sizes:
-        cascades = simulate_cascades(i, g, source=source)
-        x, y = build_matrix(cascades, 2)
-        e = infer(x, y)
-        print e
+    thresh = np.arange(0., 1.1, step=0.2)
+    sizes = np.arange(10, 100, step=10)
+    nsimul = 10
+    r = np.zeros(len(sizes), len(thresh))
+    for t in thresh:
+        for i in nsimul:
+            cascades = simulate_cascades(np.max(sizes), g,
+                                         source=lambda graph: source(graph, t))
+            e = np.zeros(g.shape[0])
+            for j, s in enumerate(sizes):
+                x, y = build_matrix(cascades, 2)
+                e += infer(x[:s], y[:s])
+
+    for i, t in enumerate(thresh):
+        plt.plot(sizes, m[:, i], label=str(t))
+    plt.legend()
+    plt.show()
+
 
         # conf = bootstrap(x, y, n_iter=100)
         # estimand = np.linalg.norm(np.delete(conf - g[0], 0, axis=1), axis=1)