diff options
author | tlacombe <lacombe1993@gmail.com> | 2020-03-17 10:55:14 +0100 |
---|---|---|
committer | tlacombe <lacombe1993@gmail.com> | 2020-03-17 10:55:14 +0100 |
commit | cdc57712ca159f3044453cef41e31ebc03617a1b (patch) | |
tree | 671e6527ac7c0b3e3ee2a0d14f1d232d97634699 /src/python/test/test_wasserstein_barycenter.py | |
parent | 2de9709b63045c484aa1c53f72c870eb210880d9 (diff) |
removed _optimal_matching from barycenter as it is now handled by wasserstein_distance.
Diffstat (limited to 'src/python/test/test_wasserstein_barycenter.py')
-rwxr-xr-x | src/python/test/test_wasserstein_barycenter.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/test/test_wasserstein_barycenter.py b/src/python/test/test_wasserstein_barycenter.py index 5167cb84..4d18616b 100755 --- a/src/python/test/test_wasserstein_barycenter.py +++ b/src/python/test/test_wasserstein_barycenter.py @@ -38,7 +38,7 @@ def test_lagrangian_barycenter(): assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg7], verbose=False) - dg7) < eps Y, log = lagrangian_barycenter(pdiagset=[dg4, dg8], verbose=True) assert np.linalg.norm(Y - np.array([[1,3], [5, 7]])) < eps - assert np.abs(log["energy"] - 4) < eps + assert np.abs(log["energy"] - 2) < eps assert np.array_equal(log["groupings"][0] , np.array([[0, -1], [1, -1]])) assert np.array_equal(log["groupings"][1] , np.array([[0, 0], [1, 1]])) assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg8, dg4], init=np.array([[0.2, 0.6], [0.5, 0.7]]), verbose=False) - np.array([[1, 3], [5, 7]])) < eps |