summaryrefslogtreecommitdiff
path: root/ot/da.py
AgeCommit message (Collapse)Author
2023-02-28[MRG] Add method argument to sinkhorn Transport (#440)tgnassou
* add method argument to sinkhron transport' * update release
2023-01-16[MRG] OT for Gaussian distributions (#428)tgnassou
* add gaussian modules * add gaussian modules * add PR to release.md * Apply suggestions from code review Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Apply suggestions from code review Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org> * Update ot/gaussian.py * Update ot/gaussian.py * add empirical bures wassertsein distance, fix docstring and test * update to fit with new networkx API * add test for jax et tf" * fix test * fix test? * add empirical_bures_wasserstein_mapping * fix docs * fix doc * fix docstring * add tgnassou to contributors * add more coverage for gaussian.py * add deprecated function * fix doc math" " * fix doc math" " * add remi flamary to authors of gaussiansmodule * fix equation Co-authored-by: Rémi Flamary <remi.flamary@gmail.com> Co-authored-by: Alexandre Gramfort <alexandre.gramfort@m4x.org>
2022-12-06[MRG] Fix bug in regularized OTDA l1lp with log (#413)Rémi Flamary
* correct bug in DA l1lp with log * better tests and speedup with smaller dataset size * remove jax for log test * remove trndorflow for log test * pep8!
2022-03-24[MRG] Domain adaptation and unbalanced solvers with backend support (#343)Nathan Cassereau
* First draft * Add matrix inverse and square root to backend * Eigen decomposition for older versions of pytorch (1.8.1 and older) * Corrected eigen decomposition for pytorch 1.8.1 and older * Spectral theorem is a thing * Optimization * small optimization * More functions converted * pep8 * remove a warning and prepare torch meshgrid for future torch release (which will change default indexing) * dots and pep8 * Meshgrid corrected for older version and prepared for future versions changes * New backend functions * Base transport * LinearTransport * All transport classes + pep8 * PR added to release file * Jcpot barycenter test * unbalanced with backend * pep8 * bug solve * test of domain adaptation with backends * solve bug for tic toc & macos * solving scipy deprecation warning * solving scipy deprecation warning attempt2 * solving scipy deprecation warning attempt3 * A warning is triggered when a float->int conversion is detected * bug solve * docs * release file updated * Better handling of float->int conversion in EMD * Corrected test for is_floating_point * docs * release file updated * cupy does not allow implicit cast * fromnumpy * added test * test da tf jax * test unbalanced with no provided histogram * using type_as argument in unif function correctly * pep8 * transport plan cast in emd changed behaviour, now trying to cast as histogram's dtype, defaulting to cost matrix Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
2021-12-09[MRG] Tensorflow backend & Benchmarker & Myst_parser (#316)Nathan Cassereau
* First batch of tf methods (to be continued) * Second batch of method (yet to debug) * tensorflow for cpu * add tf requirement * pep8 + bug * small changes * attempt to solve pymanopt bug with tf2 * attempt #2 * attempt #3 * attempt 4 * docstring * correct pep8 violation introduced in merge conflicts resolution * attempt 5 * attempt 6 * just a random try * Revert "just a random try" This reverts commit 8223e768bfe33635549fb66cca2267514a60ebbf. * GPU tests for tensorflow * pep8 * attempt to solve issue with m2r2 * Remove transpose backend method * first draft of benchmarker (need to correct time measurement) * prettier bench table * Bitsize and prettier device methods * prettified table bench * Bug corrected (results were mixed up in the final table) * Better perf counter (for GPU support) * pep8 * EMD bench * solve bug if no GPU available * pep8 * warning about tensorflow numpy api being required in the backend.py docstring * Bug solve in backend docstring * not covering code which requires a GPU * Tensorflow gradients manipulation tested * Number of warmup runs is now customizable * typo * Remove some warnings while building docs * Change prettier_device to device_type in backend * Correct JAX mistakes preventing to see the CPU if a GPU is present * Attempt to solve JAX bug in case no GPU is found * Reworked benchmarks order and results storage & clear GPU after usage by benchmark * Add bench to backend docstring * better benchs * remove useless stuff * Better device_type * Now using MYST_PARSER and solving links issue in the README.md / online docs
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-04-19[MRG] Cleanup test warnings (#242)Rémi Flamary
* remove warnings in tests from docstrings * working tets for bregman implemneted methods * pep8
2020-08-24Fix to BaseTransport.transform_labels() (#208)Samarth Mishra
* Fix to BaseTransport.transform_labels() Issue #207 * Fix - forgot to commit
2020-04-24pep8Rémi Flamary
2020-04-24add ferradans to da classes documentationRémi Flamary
2020-04-24better documentationRémi Flamary
2020-04-20added defaults for emd_laplaceievred
2020-04-20added defaults for emd_laplaceievred
2020-04-20sim+sim param fixedievred
2020-04-20added kwargs to sim + docievred
2020-04-17added regulrization from [6]+fix other issuesievred
2020-04-16references addedievred
2020-04-15import laplacianievred
2020-04-15Merge branch 'master' into laplace_daIevgen Redko
2020-04-15check conflict daievred
2020-04-15fix soft labels, remove gammas from jcpotievred
2020-04-08added label normalization to utilsievred
2020-04-08add label prop + inverseievred
2020-04-08remove blank lineievred
2020-04-08remove jcpot from laplaceievred
2020-04-08test+utils+readmeievred
2020-04-08remove laplace from jcpotievred
2020-04-03autopep+remove sinkhorn+add simtypeievred
2020-04-03v2 laplace emd sinkhornievred
2020-04-01fix imports remove checksievred
2020-03-31readme move to bregmanievred
2020-03-31v1 jcpot example testievred
2020-03-31added jcpotievred
2019-08-28correct call of unbalanced with reg_mHicham Janati
2019-08-12fixes for travis, added test, minor nitsngayraud
2019-08-12Added Unbalaced transport to domain adaptation methods. Corrected small bug ↵ngayraud
related to warnings in unbalaced.py . Added an error message when user wants to normalize with other than expected cost normalization functions.
2019-06-06start documentation otRémi Flamary
2019-05-14Fixed pep8 related issuengayraud
2019-05-10Fixed multiple docstring issuesngayraud
2018-05-29remove OTDA + pep8Rémi Flamary
2018-05-09Merge pull request #42 from rflamary/linear_mappingRémi Flamary
Linear mapping + tests
2018-05-09doc typos in linear map functionRémi Flamary
2018-04-18BUG: correct typo problemsTardy Benjamin
2018-04-18BUG: typo error lo->logTardy Benjamin
2018-04-18BUG: Parameter log unusable in sinkhorn classesTardy Benjamin
2018-04-18ENH: Change the parameter type to bool in class EMDTransport documentationTardy Benjamin
2018-04-18BUG: EMDTransport parameter log unusableTardy Benjamin
2018-03-21pep cleanupRémi Flamary
2018-03-21update readme + docRémi Flamary
2018-03-20passing testsRémi Flamary