greedy prediction

1 files changed, 11 insertions, 3 deletions

diff --git a/jpa_test/cascade_creation.py b/jpa_test/cascade_creation.py
index f8defc7..ca48431 100644
--- a/jpa_test/cascade_creation.py
+++ b/jpa_test/cascade_creation.py
@@ -47,11 +47,19 @@ def icc_cascade(G, p_init):
     return cascade
 
 
-def concat_cascades(cascades):
-    """
-    Concatenate list of cascades into matrix
+def generate_cascades(G, p_init, n_cascades):
     """
+    returns list of cascades
+    """""
+    return [icc_cascade(G,p_init) for i in xrange(n_cascades)]
+
 
+def greedy_prediction(G, cascades):
+    """
+    returns estimated graph
+    """
+    G_hat = InfluenceGraph(max_proba=None)
+    G.add_nodes_from(G.nodes())
 
 def test():
     """