summaryrefslogtreecommitdiff
path: root/test/test_ot.py
AgeCommit message (Collapse)Author
2021-11-02[MRG] Sliced and 1D Wasserstein distances : backend versions (#256)Nicolas Courty
* add numpy and torch backends * stat sets on functions * proper import * install recent torch on windows * install recent torch on windows * now testing all functions in backedn * add jax backedn * clenaup windowds * proper convert for jax backedn * pep8 * try again windows tests * test jax conversion * try proper widows tests * emd fuction ses backedn * better test partial OT * proper tests to_numpy and teplate Backend * pep8 * pep8 x2 * feaking sinkhorn works with torch * sinkhorn2 compatible * working ot.emd2 * important detach * it should work * jax autodiff emd * pep8 * no tast same for jax * new independat tests per backedn * freaking pep8 * add tests for gradients * deprecate ot.gpu * worging dist function * working dist * dist done in backedn * not in * remove indexing * change accuacy for jax * first pull backend * projection simplex * projection simplex * projection simplex * projection simplex no ci * projection simplex no ci * projection simplex no ci * pep8 * add backedn discusion to quickstart guide * projection simplex no ci * projection simplex no ci * projection simplex no ci * pep8 + better doc * proper links * corect doctest * big debug documentation * doctest again * doctest again bis * doctest again ter (last one or i kill myself) * backend test + doc proj simplex * correction test_utils * correction test_utils * correction cumsum * correction flip * correction flip v2 * more debug * more debug * more debug + pep8 * pep8 * argh * proj_simplex * backedn works for sort * proj simplex * jax sucks * update doc * Update test/test_utils.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/quickstart.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/quickstart.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/quickstart.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/readme.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update test/test_utils.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update ot/utils.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/readme.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update ot/lp/__init__.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * begin comment alex * comment alex part 2 * optimize test gromov * proj_simplex on vectors * add awesome gradient decsnt example on the weights * pep98 of course * proof read example by alex * pep8 again * encoding oos in translation * correct legend * new backend functions for sliced * small indent pb * Optimized backendversion of sliced W * error in sliced W * after master merge * error sliced * error sliced * pep8 * test_sliced pep8 * doctest + precision for sliced * doctest * type win test_backend gather * type win test_backend gather * Update sliced.py change argument of padding pad_width * Update backend.py update redefinition * Update backend.py pep8 * Update backend.py pep 8 again.... * pep8 * build docs * emd2_1D example * refectoring emd_1d and variants * remove unused previous wasserstein_1d * pep8 * upate example * move stuff * tesys should work + implemù random backend * test random generayor functions * correction * better random generation * update sliced * update sliced * proper tests sliced * max sliced * chae file nam * add stuff * example sliced flow and barycenter * correct typo + update readme * exemple sliced flow done * pep8 * solver1d works * pep8 Co-authored-by: Rémi Flamary <remi.flamary@gmail.com> Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org>
2021-10-25[MRG] Backend for optim (#282)ncassereau-idris
* Backend for optim * Bug solve * Doc update * backend tests now with fixture * Unused imports removed * Docs * Docs * Docs * Outer product backend docs * Prettier docs * Pep8 * Mistakes corrected Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
2021-06-01[MRG] POT numpy/torch/jax backends (#249)Rémi Flamary
* add numpy and torch backends * stat sets on functions * proper import * install recent torch on windows * install recent torch on windows * now testing all functions in backedn * add jax backedn * clenaup windowds * proper convert for jax backedn * pep8 * try again windows tests * test jax conversion * try proper widows tests * emd fuction ses backedn * better test partial OT * proper tests to_numpy and teplate Backend * pep8 * pep8 x2 * feaking sinkhorn works with torch * sinkhorn2 compatible * working ot.emd2 * important detach * it should work * jax autodiff emd * pep8 * no tast same for jax * new independat tests per backedn * freaking pep8 * add tests for gradients * deprecate ot.gpu * worging dist function * working dist * dist done in backedn * not in * remove indexing * change accuacy for jax * first pull backend * projection simplex * projection simplex * projection simplex * projection simplex no ci * projection simplex no ci * projection simplex no ci * pep8 * add backedn discusion to quickstart guide * projection simplex no ci * projection simplex no ci * projection simplex no ci * pep8 + better doc * proper links * corect doctest * big debug documentation * doctest again * doctest again bis * doctest again ter (last one or i kill myself) * backend test + doc proj simplex * correction test_utils * correction test_utils * correction cumsum * correction flip * correction flip v2 * more debug * more debug * more debug + pep8 * pep8 * argh * proj_simplex * backedn works for sort * proj simplex * jax sucks * update doc * Update test/test_utils.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/quickstart.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/quickstart.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/quickstart.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/readme.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update test/test_utils.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update ot/utils.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update docs/source/readme.rst Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update ot/lp/__init__.py Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * begin comment alex * comment alex part 2 * optimize test gromov * proj_simplex on vectors * add awesome gradient decsnt example on the weights * pep98 of course * proof read example by alex * pep8 again * encoding oos in translation * correct legend Co-authored-by: Nicolas Courty <ncourty@irisa.fr> Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org>
2021-04-08[MRG] ADD JMLR paper to the readme and documentation (#231)Rémi Flamary
* add JMLR reefrence to eradme and doc
2020-04-23[WIP] Issue with sparse emd and adding tests on macos (#158)Nicolas Courty
* First commit-warning removal * remove dense feature * pep8 * pep8 * EMD.h * pep8 again * tic toc tolerance Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
2020-04-02Fix testAdrienCorenflos
2020-04-01Fix orderingAdrienCorenflos
2020-01-30comlete documentationRémi Flamary
2020-01-27correct pep8Rémi Flamary
2020-01-27add test for constraint viuolation of dualsRémi Flamary
2020-01-07Merge branch 'master' into emd_dimensionKilian
2019-12-18update testRémi Flamary
2019-12-09pep8Kilian
2019-12-09calling ot.emd testKilian
2019-12-09add assert for emd dimension mismatchKilian
2019-12-02pep8Rémi Flamary
2019-12-02add test emd2 multiRémi Flamary
2019-12-02add test emd2Rémi Flamary
2019-12-02add test emd2Rémi Flamary
2019-12-02remove useless variableRémi Flamary
2019-12-02proper documentation and parameterRémi Flamary
2019-12-02add awesome sparse solverRémi Flamary
2019-07-03test raise with pytets in test_emd_1d_emd2_1dRémi Flamary
2019-07-01Explicit doctest call in travis + removed uneffective doctest in test_otRomain Tavenard
2019-06-27Wasserstein defined as the cost itself (do not return transportation matrix)Romain Tavenard
2019-06-27Improved tests and docs for wasserstein_1dRomain Tavenard
2019-06-21Started documentingRomain Tavenard
2019-06-21Sparse G matrix for EMD1d + standard metrics computed without cdistRomain Tavenard
2019-06-20EMD 1d without docRomain Tavenard
2018-09-28speedup testsRémi Flamary
2018-07-09add test free support barycenter algorithm + cleaningVivien Seguy
2018-07-09add test free support barycenter algorithm + cleaningVivien Seguy
2018-05-30replace function name tin testsRémi Flamary
2018-05-11add test barycenter cvxoptRémi Flamary
2018-05-11add test for barycenterRémi Flamary
2017-09-12Made the return of the matrix optional in emd2Antoine Rolet
2017-09-09Returned to old variable name to follow repo conventionAntoine Rolet
2017-09-09Renamed variable in stringAntoine Rolet
2017-09-09Renamed variableAntoine Rolet
2017-09-09Renamed variablesAntoine Rolet
2017-09-07Removed unused variable declarationAntoine Rolet
2017-09-07Added log option to muliprocess emdAntoine Rolet
2017-09-07Return dual variables in an optional dictionaryAntoine Rolet
Also removed some code duplication
2017-09-07Added MAX_ITER_REACHED flag and warningAntoine Rolet
2017-09-05ReformatAntoine Rolet
2017-09-05Fixed typo and merged emd testsAntoine Rolet
2017-07-26author and license for tets filesRémi Flamary
2017-07-26flake8Rémi Flamary
2017-07-26test_ot random stateRémi Flamary
2017-07-26clean test_otRémi Flamary
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-30 09:40:46 +0000