-
Notifications
You must be signed in to change notification settings - Fork 506
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[MRG] Generalized Wasserstein Barycenters #372
Conversation
Codecov Report
@@ Coverage Diff @@
## master #372 +/- ##
==========================================
+ Coverage 93.78% 93.85% +0.07%
==========================================
Files 23 23
Lines 5902 5926 +24
==========================================
+ Hits 5535 5562 +27
+ Misses 367 364 -3 |
@@ -137,6 +137,7 @@ def test_partial_wasserstein(): | |||
|
|||
|
|||
def test_partial_gromov_wasserstein(): | |||
np.random.seed(42) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@eloitanguy @rflamary you should not use global seeds in tests. It makes tests coupled as they share a random state. It may happen that a test that does not control its seed fails only if run after another test and it makes debugging a nightmare. You should np.random.default_rng these days.
Types of changes
generalized_free_support_barycenter
solver inot/lp/__init__.py
(next tofree_support_barycenter
)examples/barycenters/plot_generalized_free_support_barycenter.py
free_support_barycenter
(mis-numbered/missing references + added details on variable names)Motivation and context / Related issue
Implements the article "Generalised Wasserstein Barycenter" [42]
[42] DELON, Julie, GOZLAN, Nathael, et SAINT-DIZIER, Alexandre.
Generalized Wasserstein barycenters between probability measures living on different subspaces.
arXiv preprint arXiv:2105.09755, 2021.
How has this been tested (if it applies)
Added tests in
tests_ot.py
, accounting for dimensionality, optional arguments and backendsPR checklist