From fd0b6eefd851dbe7528383209e0041afeb6766af Mon Sep 17 00:00:00 2001 From: GeoffNN Date: Fri, 19 Jun 2020 14:44:45 +0200 Subject: [PATCH] Added sublinear stepsize --- examples/frank_wolfe/plot_vertex_overlap.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/frank_wolfe/plot_vertex_overlap.py b/examples/frank_wolfe/plot_vertex_overlap.py index 37306d0f..7fae6988 100755 --- a/examples/frank_wolfe/plot_vertex_overlap.py +++ b/examples/frank_wolfe/plot_vertex_overlap.py @@ -38,7 +38,8 @@ x0 = np.zeros(n_features) for i, (step, label, marker) in enumerate( - [["backtracking", "backtracking", "^"], ["DR", "DR step-size", "d"]] + [["backtracking", "backtracking", "^"], ["DR", "DR step-size", "d"], + ["sublinear", "sublinear", "s"]] ): print("Running %s variant" % label) st_prev = [] @@ -66,7 +67,7 @@ def trace(kw): x0, l1_ball.lmo, callback=trace, - max_iter=int(1e4), + max_iter=int(1e2), step=step, verbose=True, lipschitz=f.lipschitz,