diff options
author | Kilian Fatras <kilianfatras@Kilians-MacBook-Air.local> | 2019-03-29 13:44:07 +0100 |
---|---|---|
committer | Kilian Fatras <kilianfatras@Kilians-MacBook-Air.local> | 2019-03-29 13:44:07 +0100 |
commit | f63712f4df213bbfe0a2665390d51974305de705 (patch) | |
tree | aa72d02f33903db3c3801f93f68c6efa8a719325 /ot/bregman.py | |
parent | d754a645f9b4ef88d7e0aba1188fa83d7d58af1f (diff) |
call ot.unif and ot.dist
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index f6aa339..9e9989f 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -1375,9 +1375,9 @@ def empirical_sinkhorn(X_s, X_t, reg, a=None, b=None, metric='sqeuclidean', numI ''' if a is None: - a = utils.unif(np.shape(X_s)[0]) + a = ot.unif(np.shape(X_s)[0]) if b is None: - b = utils.unif(np.shape(X_t)[0]) + b = ot.unif(np.shape(X_t)[0]) M = ot.dist(X_s, X_t, metric=metric) @@ -1465,9 +1465,9 @@ def empirical_sinkhorn2(X_s, X_t, reg, a=None, b=None, metric='sqeuclidean', num ''' if a is None: - a = utils.unif(np.shape(X_s)[0]) + a = ot.unif(np.shape(X_s)[0]) if b is None: - b = utils.unif(np.shape(X_t)[0]) + b = ot.unif(np.shape(X_t)[0]) M = ot.dist(X_s, X_t, metric=metric) |