From ee58bbeb5e8d870835c03d7f36b9976a0e8b857f Mon Sep 17 00:00:00 2001 From: jeanpouget-abadie Date: Fri, 5 Dec 2014 12:19:45 -0500 Subject: timeout decorator better usage --- jpa_test/algorithms.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'jpa_test/algorithms.py') diff --git a/jpa_test/algorithms.py b/jpa_test/algorithms.py index f3bf917..0d5f154 100644 --- a/jpa_test/algorithms.py +++ b/jpa_test/algorithms.py @@ -34,7 +34,7 @@ def greedy_prediction(G, cascades): unaccounted[t] = False return G_hat -@timeout.timeout(10) + def recovery_l1obj_l2constraint(G, cascades): """ Returns estimated graph from following convex program: @@ -45,10 +45,13 @@ def recovery_l1obj_l2constraint(G, cascades): G_hat.add_nodes_from(G.nodes()) for node in G_hat.nodes(): print node - M, w = cascade_creation.icc_matrixvector_for_node(cascades, node) - p_node, __ = convex_optimization.l1obj_l2constraint(M,w) - G_hat = cascade_creation.add_edges_from_proba_vector(G=G_hat, - p_node=p_node, node=node, floor_cstt=.01) + try: + M, w = cascade_creation.icc_matrixvector_for_node(cascades, node) + p_node, __ = convex_optimization.l1obj_l2constraint(M,w) + G_hat = cascade_creation.add_edges_from_proba_vector(G=G_hat, + p_node=p_node, node=node, floor_cstt=.01) + except timeout.TimeoutError: + print "TimeoutError, skipping to next node" return G_hat @@ -76,7 +79,7 @@ def test(): G.erdos_init(n = 100, p = .3) import time t0 = time.time() - A = cascade_creation.generate_cascades(G, .2, 50) + A = cascade_creation.generate_cascades(G, .2, 100) G_hat = recovery_l1obj_l2constraint(G, A) -- cgit v1.2.3-70-g09d2