[MRG] BAPG solvers for GW and FGW by cedricvincentcuaz · Pull Request #581 · PythonOT/POT
Types of changes
-
Add Bregman alternated projected gradient solvers for GW and FGW via functions:
BAPG_gromov_wasserstein,BAPG_gromov_wasserstein2,BAPG_fused_gromov_wasserstein,BAPG_fused_gromov_wasserstein2 -
update example with FGW solvers comparison.
Motivation and context / Related issue
- Keep up with the (F)GW literature.
How has this been tested (if it applies)
- add tests
test_BAPG_fgw,test_BAPG_gromov. Modified in placetest_entropic_gromov_dtype_device,test_entropic_fgw_dtype_device.
PR checklist
- I have read the CONTRIBUTING document.
- The documentation is up-to-date with the changes I made (check build artifacts).
- All tests passed, and additional code has been covered with new tests.
- I have added the PR and Issue fix to the RELEASES.md file.