summaryrefslogtreecommitdiff
path: root/ot/gromov.py
AgeCommit message (Collapse)Author
2021-11-04[MRG] Make gromov loss differentiable wrt matrices and weights (#302)Rémi Flamary
* grmov differentable * new stuff * test gromov gradients * fgwdifferentiable * fgw tested * correc name test * add awesome example with gromov optimizatrion * pep8+ typos * damn pep8 * thunbnail * remove prints
2021-11-03[MRG] Docs updates (#298)ncassereau-idris
* bregman docs * sliced docs * docs partial * unbalanced docs * stochastic docs * plot docs * datasets docs * utils docs * dr docs * dr docs corrected * smooth docs * docs da * pep8 * docs gromov * more space after min and argmin * docs lp * bregman docs * bregman docs mistake corrected * pep8 Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
2021-11-02[MRG] Backend for gromov (#294)ncassereau-idris
* bregman: small correction * gromov backend first draft * Removing decorators * Reworked casting method * Bug solve * Removing casting * Bug solve * toarray renamed todense ; expand_dims removed * Warning (jax not supporting sparse matrix) moved * Mistake corrected * test backend * Sparsity test for older versions of pytorch * Trying pytorch/1.10 * Attempt to correct torch sparse bug * Backend version of gromov tests * Random state introduced for remaining gromov functions * review changes * code coverage * Docs (first draft, to be continued) * Gromov docs * Prettified docs * mistake corrected in the docs * little change Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
2021-10-27[WIP] Sinkhorn in log space (#290)Rémi Flamary
* adda sinkhorn log and working sinkhorn2 function * more tests pass * more tests pass * it works but not by default yet * remove warningd * update circleci doc * update circleci doc * new sinkhorn implemeted but not by default * better * doctest pass * test doctest * new test utils * remove pep8 errors * remove pep8 errors * doc new implementtaion with log * test sinkhorn 2 * doc for log implementation
2021-10-25[MRG] Bregman backend (#280)ncassereau-idris
* Bregman * Resolve conflicts * Bug solve * Bregman updated for JAX compatibility * Tests coherence between backend improved * No longer enforcing 64 bits operations on Jax except for tests * Now using mixtures, to make backend dependent tests with less code * Better test skipping code * Pep8 + test optimizations * redundancy removed * Docs * Typo corrected * Typo * Typo * Docs * Docs * pep8 * Backend docs * Prettier docs * Mistake corrected * small changes * Better wording Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
2021-09-17[MRG] Implementation of two news algorithms: SaGroW and PoGroW. (#275)Tanguy
* Add two new algorithms to solve Gromov Wasserstein: Sampled Gromov Wasserstein and Pointwise Gromov Wasserstein. * Correct some lines in SaGroW and PoGroW to follow pep8 guide. * Change nb_samples name. Use rdm state. Change symmetric check. * Change names of len(p) and len(q) in SaGroW and PoGroW. * Re-add some deleted lines in the comments of gromov.py Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
2021-04-19[MRG] Cleanup test warnings (#242)Rémi Flamary
* remove warnings in tests from docstrings * working tets for bregman implemneted methods * pep8
2020-04-24pep8Rémi Flamary
2020-04-24better doc per moduleRémi Flamary
2020-03-10fixing trailing and before arithmetic operation whitespace issuesNemo Fournier
2020-03-09clean and complete the document of fgw related functionsNemo Fournier
2020-03-09fix fgw alpha parameter implementationNemo Fournier
2019-11-13fix log bug in gromov_wasserstein2Kilian
2019-07-09first pass with adding pydocstyle in makefileAlexandre Gramfort
2019-06-25alpha for documentationRémi Flamary
2019-06-25pep8Rémi Flamary
2019-06-25cleanup documentation gromovRémi Flamary
2019-06-04changes forgotten comentstvayer
2019-05-29code review1tvayer
2019-05-29Doc+armijotvayer
2019-05-29test checktvayer
2019-05-29Merge remote-tracking branch 'rflamary/master'tvayer
merge pot
2019-05-29pep8tvayer
2019-05-29Tests + contributionstvayer
2019-05-28solve conlicttvayer
2019-05-28bary fgwtvayer
2019-05-28FGW+gromov changestvayer
2019-03-15correction doc in gromov.pyRémi Flamary
2018-05-30update examples and notebooksRémi Flamary
2018-03-21add entropic gromov test for 90+% corerageRémi Flamary
2018-03-21add externals for function signatureRémi Flamary
2018-02-16should pass tests nowRémi Flamary
2018-02-16better pep8 but not solvedRémi Flamary
2018-02-16update examplesRémi Flamary
2018-02-15awesome documentationRémi Flamary
2018-02-15add documentationRémi Flamary
2018-02-15large update gromovRémi Flamary
2017-09-13Corrections on Gromovncourty
2017-09-13Corrections on GromovNicolas Courty
2017-09-12Corrections on GromovNicolas Courty
2017-09-01docstrings + namingNicolas Courty
2017-09-01docstrings + namingNicolas Courty
2017-09-01solving conflicts :/Nicolas Courty
2017-09-01docstrings and namingNicolas Courty
2017-09-01docstrings and namingNicolas Courty
2017-09-01Minor corrections suggested by @agramfort + new barycenter example + test ↵Nicolas Courty
function
2017-09-01gromov:flake8 and otherNicolas Courty
2017-09-01Gromov-Wasserstein distanceNicolas Courty
2017-08-31Minor corrections suggested by @agramfort + new barycenter example + test ↵Nicolas Courty
function
2017-08-28gromov:flake8 and otherNicolas Courty