Skip to content

Commit cd193f7

Browse files
author
Kilian Fatras
committed
updated README and fixed pep8
1 parent b13feb0 commit cd193f7

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

test/test_stochastic.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -193,17 +193,14 @@ def test_dual_sgd_sinkhorn():
193193

194194
# Test gaussian
195195
n = 30
196-
n_source = n
197-
n_target = n
198196
reg = 1
199-
numItermax = 150000
200197
batch_size = 30
201198

202-
a = ot.datasets.get_1D_gauss(n_source, m=15, s=5) # m= mean, s= std
203-
b = ot.datasets.get_1D_gauss(n_target, m=15, s=5)
204-
X_source = np.arange(n_source,dtype=np.float64)
205-
Y_target = np.arange(n_target,dtype=np.float64)
206-
M = ot.dist(X_source.reshape((n_source, 1)), Y_target.reshape((n_target, 1)))
199+
a = ot.datasets.get_1D_gauss(n, m=15, s=5) # m= mean, s= std
200+
b = ot.datasets.get_1D_gauss(n, m=15, s=5)
201+
X_source = np.arange(n, dtype=np.float64)
202+
Y_target = np.arange(n, dtype=np.float64)
203+
M = ot.dist(X_source.reshape((n, 1)), Y_target.reshape((n, 1)))
207204
M /= M.max()
208205

209206
G_sgd = ot.stochastic.solve_dual_entropic(a, b, M, reg, batch_size,

0 commit comments

Comments
 (0)