1 files changed, 13 insertions, 0 deletions
diff --git a/jpa_test/cascade_creation.py b/jpa_test/cascade_creation.py
index c5d33cc..6e5c34e 100644
--- a/jpa_test/cascade_creation.py
+++ b/jpa_test/cascade_creation.py
@@ -66,6 +66,19 @@ def icc_cascade(G, p_init):
     return cascade
 
 
+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():
     """