POT
POT copied to clipboard
PythonOT
POT : Python Optimal Transport
Update link to contributing file ## Types of changes ## Motivation and context / Related issue ## How has this been tested (if it applies) ## PR checklist - [...
## Types of changes This PR aims to add the Translation Invariant Sinkhorn algorithm proposed in [Faster Unbalanced Optimal Transport: Translation invariant Sinkhorn and 1-D Frank-Wolfe ](https://proceedings.mlr.press/v151/sejourne22a.html) to solve the...
I am computing matrices of W_2 distances with ot.bregman.empirical_sinkhorn2 between point clouds centered along a curve. I expect that the distance from a point cloud to itself should be zero...
## Describe the bug It seems that the `cost_correction` matrix is computed incorrectly. This is the current code that can be found [here](https://github.com/PythonOT/POT/blob/47c592584d127ee01d2fa93a55cc440146fba82b/ot/da.py#L509-L518): ```python # labels_match is a (ns, nt)...
## Types of changes This PR aims to improve the current module `unbalanced.py` on two aspects: - The current module contains 3 seperate groups of unbalanced OT solvers: Sinkhorn, Majorization-Minimization...
## Types of changes - create duplicates of partial GW related functions from `ot/partial.py` to `ot/gromov/_partial.py` while adding a depreciation notes and warnings within `ot/partial.py` to maintain the API for...
## Types of changes This very large PR introduce a new style for the coe thta follow ruff formating with a fats pre-commit that will hopfully avoid "pep8" cmmts in...
## Types of changes This PR aims to add the Bures-Wasserstein gradient descent solver to compute Bures-Wasserstein barycenters (see e.g. [Gradient descent algorithms for Bures-Wasserstein barycenters](https://proceedings.mlr.press/v125/chewi20a.html) or [Averaging on the...
Unless I'm missing something, it seems the note on when not to use POT at https://pythonot.github.io/quickstart.html#when-not-to-use-pot is no longer relevant, since the geomloss functionality is now fully wrapped in POT?...
## Describe the bug I get the following error: ```python my_conda_env/python3.11/site-packages/ot/backend.py", line 1822, in __init__ self.rng_cuda_ = torch.Generator("cuda") ^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: CUDA error: initialization error ``` I am using `pot=0.9.4=py311h14de704_0` and...
