aboutsummaryrefslogtreecommitdiffstats
path: root/jpa_test/cascade_creation.py
diff options
context:
space:
mode:
Diffstat (limited to 'jpa_test/cascade_creation.py')
-rw-r--r--jpa_test/cascade_creation.py10
1 files changed, 1 insertions, 9 deletions
diff --git a/jpa_test/cascade_creation.py b/jpa_test/cascade_creation.py
index 6e5c34e..19e799f 100644
--- a/jpa_test/cascade_creation.py
+++ b/jpa_test/cascade_creation.py
@@ -26,13 +26,12 @@ class InfluenceGraph(nx.Graph):
for edge in data:
split1 = edge.split('\t')
split2 = split1[1].split('\n')
-
u = int(split1[0])
v = int(split2[0])
G.add_edge(u,v)
self.add_nodes_from(G.nodes())
self.add_edges_from(G.edges())
-
+
@property
def mat(self):
if not hasattr(self, '_mat'):
@@ -73,13 +72,6 @@ def generate_cascades(G, p_init, n_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():
"""
unit test