diff options
author | Kilian Fatras <kilianfatras@dhcp-206-12-53-210.eduroam.wireless.ubc.ca> | 2018-06-18 18:05:48 -0700 |
---|---|---|
committer | Kilian Fatras <kilianfatras@dhcp-206-12-53-210.eduroam.wireless.ubc.ca> | 2018-06-18 18:05:48 -0700 |
commit | e068b58ba4234792d96287afd34c3cddef544dd4 (patch) | |
tree | 9e03c736340fb0fab95a3880060044e58d1aa001 /examples/plot_stochastic.py | |
parent | 74cfe5ac77c3e964a85ef90c11d8ebffa16ddcfe (diff) |
pep8
Diffstat (limited to 'examples/plot_stochastic.py')
-rw-r--r-- | examples/plot_stochastic.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/plot_stochastic.py b/examples/plot_stochastic.py index 3fc1955..e139473 100644 --- a/examples/plot_stochastic.py +++ b/examples/plot_stochastic.py @@ -53,7 +53,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, lr) print(sag_pi) ############################################################################# @@ -91,7 +91,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, lr, log) print(log['alpha'], log['beta']) print(asgd_pi) @@ -174,7 +174,7 @@ M = ot.dist(X_source, Y_target) # Call ot.solve_dual_entropic and plot the results. sgd_dual_pi, log = ot.stochastic.solve_dual_entropic(a, b, M, reg, batch_size, - numItermax, lr, log) + numItermax, lr, log) print(log['alpha'], log['beta']) print(sgd_dual_pi) |