Skip to content

Commit

Permalink
[WIP] Partial Fused Gromov-Wasserstein solver (#687)
Browse files Browse the repository at this point in the history
* merge

* new dev version

* first commit partial fgw

* complete tests + solve_gromov

* complete tests + solve_gromov

* release
  • Loading branch information
cedricvincentcuaz authored Nov 18, 2024
1 parent 4435898 commit b5cfb91
Show file tree
Hide file tree
Showing 7 changed files with 746 additions and 7 deletions.
3 changes: 2 additions & 1 deletion CONTRIBUTORS.md
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,8 @@ The contributors to this library are:
* [Tanguy Kerdoncuff](https://hv0nnus.github.io/) (Sampled Gromov Wasserstein)
* [Minhui Huang](https://mhhuang95.github.io) (Projection Robust Wasserstein Distance)
* [Nathan Cassereau](https://github.com/ncassereau-idris) (Backends)
* [Cédric Vincent-Cuaz](https://github.com/cedricvincentcuaz) (Graph Dictionary Learning, FGW, semi-relaxed FGW, quantized FGW)
* [Cédric Vincent-Cuaz](https://github.com/cedricvincentcuaz) (Graph Dictionary Learning, FGW,
semi-relaxed FGW, quantized FGW, partial FGW)
* [Eloi Tanguy](https://github.com/eloitanguy) (Generalized Wasserstein
Barycenters, GMMOT)
* [Camille Le Coz](https://www.linkedin.com/in/camille-le-coz-8593b91a1/) (EMD2 debug)
Expand Down
6 changes: 4 additions & 2 deletions RELEASES.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,11 @@

## 0.9.6dev

#### New features
- Implement CG solvers for partial FGW (PR #687)

#### Closed issues
- Fixed `ot.mapping` solvers which depended on deprecated `cvxpy` `ECOS` solver
(PR #692, Issue #668)
- Fixed `ot.mapping` solvers which depended on deprecated `cvxpy` `ECOS` solver (PR #692, Issue #668)


## 0.9.5
Expand Down
4 changes: 4 additions & 0 deletions ot/gromov/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,8 @@
from ._partial import (
partial_gromov_wasserstein,
partial_gromov_wasserstein2,
partial_fused_gromov_wasserstein,
partial_fused_gromov_wasserstein2,
solve_partial_gromov_linesearch,
entropic_partial_gromov_wasserstein,
entropic_partial_gromov_wasserstein2,
Expand Down Expand Up @@ -173,6 +175,8 @@
"fused_unbalanced_across_spaces_divergence",
"partial_gromov_wasserstein",
"partial_gromov_wasserstein2",
"partial_fused_gromov_wasserstein",
"partial_fused_gromov_wasserstein2",
"solve_partial_gromov_linesearch",
"entropic_partial_gromov_wasserstein",
"entropic_partial_gromov_wasserstein2",
Expand Down
Loading

0 comments on commit b5cfb91

Please sign in to comment.