POT

POT copied to clipboard

[MRG] Linear Circular OT

1

## Types of changes This PR aims to add the Linear Circular OT distance (with uniform measure on the circle as reference measure) introduced in [LCOT: Linear Circular Optimal Transport...

clbonet

[MRG] OT barycenters for generic transport costs

3

## Types of changes - [x] contribute a generalisation of `free_support` that accepts any cost function (implements [this paper](https://arxiv.org/abs/2501.04016)) - [x] contribute a specific version of this function in `ot.gmm`...

eloitanguy

[WIP] Wrappers for OT barycenter solvers

1

## Types of changes - creation of the class `ot.utils.BaryResult` extending `OTResult` first draft for OT barycenter solvers with free support and predefined metric `ot.solvers.bary_free_support` built on top of `ot.solvers.solve_sample`,...

cedricvincentcuaz

typo in \text{max_nz} in ot.utils.projection_sparse_simplex(

2

## Types of changes typo correction for doc display. ## PR checklist - [x] I have read the [**CONTRIBUTING**](https://pythonot.github.io/contributing.html) document. - [x] The documentation is up-to-date with the changes I...

josephsalmon

[MRG] Fix None init plan in unbalanced lbfgs solvers

1

## Types of changes fix undesirable behaviour of `ot.unbalanced.lbfgsb_unbalanced` when `isinstance(reg_div, tuple)` to test whether included functions in reg_div can handle numpy arrays. These tests are now handled via `ot.utils.fun_to_numpy`....

cedricvincentcuaz

[WIP] Add a stablized function entropic_partial_wasserstein_logscale

4

1. add a new function called entropic\_partial\_wasserstein\_logscale to ot.partial. This function solves the same problem as entropic\_partial\_wasserstein but is computed in logscale, so it is more robust. 2. Test exampless...

wzm2256

Wasserstein Circle distance doesn't seem correct?

4

## Describe the bug Forgive me if I've misunderstood what the `wasserstein_circle` function is supposed to do. But I would have thought that if we have ```python d1 = wasserstein_circle(arr1,...

ckp95

[WIP] implementation of low rank ot via factor relaxation paper

4

## Types of changes Added implementation of the paper Low-Rank Optimal Transport through Factor Relaxation with Latent Coupling https://arxiv.org/abs/2411.10555 ## How has this been tested (if it applies) Tested it...

fegounna

entropic_partial_wasserstein not stable

1

## Describe the bug The entropic_partial_wasserstein function produces nan when eps is small. ### To Reproduce ``` import ot import torch import numpy as np def compute_OT(M, alpha, beta, epsilon):...

wzm2256

[WIP] Sparse implementation of EMD

1

The goal is to return a sparse transport in `ot.emd` plan when the cost matrix matrix is itself sparse which corresponds to infinite cost between points, or when no value...

tvayer