diff options
| author | Kilian Fatras <kilianfatras@dhcp-206-12-53-92.eduroam.wireless.ubc.ca> | 2018-06-21 17:18:26 -0700 |
|---|---|---|
| committer | Kilian Fatras <kilianfatras@dhcp-206-12-53-92.eduroam.wireless.ubc.ca> | 2018-06-21 17:18:26 -0700 |
| commit | 6777ffd5c8457faac4467e58ba9edbcf2f86961b (patch) | |
| tree | 286c1fdb3902031d47cc5e1bdefab97ab9c9da22 /examples | |
| parent | 7073e417bed151976c62fe20d1ba69abd30e7758 (diff) | |
gave better step size ASGD & SAG
Diffstat (limited to 'examples')
| -rw-r--r-- | examples/plot_stochastic.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/examples/plot_stochastic.py b/examples/plot_stochastic.py index 09b95d0..6274b4c 100644 --- a/examples/plot_stochastic.py +++ b/examples/plot_stochastic.py @@ -32,8 +32,7 @@ print("------------SEMI-DUAL PROBLEM------------") n_source = 7 n_target = 4 reg = 1 -numItermax = 10000 -lr = 0.1 +numItermax = 1000 a = ot.utils.unif(n_source) b = ot.utils.unif(n_target) @@ -53,7 +52,7 @@ M = ot.dist(X_source, Y_target) method = "SAG" sag_pi = ot.stochastic.solve_semi_dual_entropic(a, b, M, reg, method, - numItermax, lr) + numItermax) print(sag_pi) ############################################################################# @@ -68,8 +67,7 @@ print(sag_pi) n_source = 7 n_target = 4 reg = 1 -numItermax = 100000 -lr = 1 +numItermax = 1000 log = True a = ot.utils.unif(n_source) @@ -91,7 +89,7 @@ M = ot.dist(X_source, Y_target) method = "ASGD" asgd_pi, log = ot.stochastic.solve_semi_dual_entropic(a, b, M, reg, method, - numItermax, lr, log) + numItermax, log) print(log['alpha'], log['beta']) print(asgd_pi) |