diff options
Diffstat (limited to 'jpa_test/algorithms.py')
| -rw-r--r-- | jpa_test/algorithms.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/jpa_test/algorithms.py b/jpa_test/algorithms.py index 2ee08ec..9973dc3 100644 --- a/jpa_test/algorithms.py +++ b/jpa_test/algorithms.py @@ -8,13 +8,10 @@ def greedy_prediction(G, cascades): returns estimated graph """ G_hat = cascade_creation.InfluenceGraph(max_proba=None) - G.add_nodes_from(G.nodes()) + G_hat.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(): """ |