@@ -199,13 +199,13 @@ def test_empirical_sinkhorn():
199
199
X_s = np .reshape (np .arange (n ), (n , 1 ))
200
200
X_t = np .reshape (np .arange (0 , n ), (n , 1 ))
201
201
M = ot .dist (X_s , X_t )
202
- M_e = ot .dist (X_s , X_t , metric = 'euclidean ' )
202
+ M_m = ot .dist (X_s , X_t , metric = 'minkowski ' )
203
203
204
204
G_sqe = ot .bregman .empirical_sinkhorn (X_s , X_t , 1 )
205
205
sinkhorn_sqe = ot .sinkhorn (a , b , M , 1 )
206
206
207
- G_e = ot .bregman .empirical_sinkhorn (X_s , X_t , 1 )
208
- sinkhorn_e = ot .sinkhorn (a , b , M_e , 1 )
207
+ G_m = ot .bregman .empirical_sinkhorn (X_s , X_t , 1 , metric = 'minkowski' )
208
+ sinkhorn_m = ot .sinkhorn (a , b , M_m , 1 )
209
209
210
210
loss_emp_sinkhorn = ot .bregman .empirical_sinkhorn2 (X_s , X_t , 1 )
211
211
loss_sinkhorn = ot .sinkhorn2 (a , b , M , 1 )
@@ -216,9 +216,9 @@ def test_empirical_sinkhorn():
216
216
np .testing .assert_allclose (
217
217
sinkhorn_sqe .sum (0 ), G_sqe .sum (0 ), atol = 1e-05 ) # metric sqeuclidian
218
218
np .testing .assert_allclose (
219
- sinkhorn_e .sum (1 ), G_e .sum (1 ), atol = 1e-05 ) # metric euclidian
219
+ sinkhorn_m .sum (1 ), G_m .sum (1 ), atol = 1e-05 ) # metric euclidian
220
220
np .testing .assert_allclose (
221
- sinkhorn_e .sum (0 ), G_e .sum (0 ), atol = 1e-05 ) # metric euclidian
221
+ sinkhorn_m .sum (0 ), G_m .sum (0 ), atol = 1e-05 ) # metric euclidian
222
222
np .testing .assert_allclose (loss_emp_sinkhorn , loss_sinkhorn , atol = 1e-05 )
223
223
224
224
0 commit comments