From 59add61ec7b8ada40909edddfb88f193e489303a Mon Sep 17 00:00:00 2001
From: jeanpouget-abadie <jean.pougetabadie@gmail.com>
Date: Sat, 29 Nov 2014 13:38:11 -0500
Subject: algorithms.py

---
 jpa_test/algorithms.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 jpa_test/algorithms.py

(limited to 'jpa_test/algorithms.py')

diff --git a/jpa_test/algorithms.py b/jpa_test/algorithms.py
new file mode 100644
index 0000000..2ee08ec
--- /dev/null
+++ b/jpa_test/algorithms.py
@@ -0,0 +1,32 @@
+import numpy as np
+import networkx as nx
+import cascade_creation
+
+
+def greedy_prediction(G, cascades):
+    """
+    returns estimated graph
+    """
+    G_hat = cascade_creation.InfluenceGraph(max_proba=None)
+    G.add_nodes_from(G.nodes())
+    for node in G.nodes():
+        unaccounted = cascades
+        for cascade in cascades:
+            time_step = 0
+            while not cascade[time_step][node]:
+                time_step += 1
+
+def test():
+    """
+    unit test
+    """
+    G = cascade_creation.InfluenceGraph(max_proba = .3)
+    G.erdos_init(n = 100, p = 1)
+    import time
+    t0 = time.time()
+    print len(cascade_creation.icc_cascade(G, p_init=.1))
+    t1 = time.time()
+    print t1 - t0
+
+if __name__=="__main__":
+    test()
-- 
cgit v1.2.3-70-g09d2