diff options
Diffstat (limited to 'jpa_test/cascade_creation.py')
| -rw-r--r-- | jpa_test/cascade_creation.py | 10 |
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 |
