summaryrefslogtreecommitdiff
path: root/ot/bregman.py
diff options
context:
space:
mode:
authorRomain Tavenard <romain.tavenard@univ-rennes2.fr>2019-07-01 12:16:04 +0200
committerRomain Tavenard <romain.tavenard@univ-rennes2.fr>2019-07-01 12:16:04 +0200
commit1fe13ed9cdda363c95c84f95fc70dcd95ac276f1 (patch)
tree92b91c79dcaa38748da1a750ab60d5ff81fbbb4c /ot/bregman.py
parentb05d315b0994d328029d4a4fc082f6994e7f06d1 (diff)
Fixed doctests
Diffstat (limited to 'ot/bregman.py')
-rw-r--r--ot/bregman.py17
1 files changed, 7 insertions, 10 deletions
diff --git a/ot/bregman.py b/ot/bregman.py
index 09716e6..8225967 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -1360,10 +1360,9 @@ def empirical_sinkhorn(X_s, X_t, reg, a=None, b=None, metric='sqeuclidean', numI
>>> reg = 0.1
>>> X_s = np.reshape(np.arange(n_s), (n_s, 1))
>>> X_t = np.reshape(np.arange(0, n_t), (n_t, 1))
- >>> emp_sinkhorn = empirical_sinkhorn(X_s, X_t, reg, verbose=False)
- >>> print(emp_sinkhorn)
- >>> [[4.99977301e-01 2.26989344e-05]
- [2.26989344e-05 4.99977301e-01]]
+ >>> empirical_sinkhorn(X_s, X_t, reg, verbose=False) # doctest: +NORMALIZE_WHITESPACE
+ array([[4.99977301e-01, 2.26989344e-05],
+ [2.26989344e-05, 4.99977301e-01]])
References
@@ -1451,9 +1450,8 @@ def empirical_sinkhorn2(X_s, X_t, reg, a=None, b=None, metric='sqeuclidean', num
>>> reg = 0.1
>>> X_s = np.reshape(np.arange(n_s), (n_s, 1))
>>> X_t = np.reshape(np.arange(0, n_t), (n_t, 1))
- >>> loss_sinkhorn = empirical_sinkhorn2(X_s, X_t, reg, verbose=False)
- >>> print(loss_sinkhorn)
- >>> [4.53978687e-05]
+ >>> empirical_sinkhorn2(X_s, X_t, reg, verbose=False)
+ array([4.53978687e-05])
References
@@ -1560,9 +1558,8 @@ def empirical_sinkhorn_divergence(X_s, X_t, reg, a=None, b=None, metric='sqeucli
>>> reg = 0.1
>>> X_s = np.reshape(np.arange(n_s), (n_s, 1))
>>> X_t = np.reshape(np.arange(0, n_t), (n_t, 1))
- >>> emp_sinkhorn_div = empirical_sinkhorn_divergence(X_s, X_t, reg)
- >>> print(emp_sinkhorn_div)
- >>> [2.99977435]
+ >>> empirical_sinkhorn_divergence(X_s, X_t, reg)
+ array([2.99977435])
References