https://github.com/pythonot/coot

CO-Optimal Transport
https://github.com/pythonot/coot

Last synced: about 1 month ago
JSON representation

CO-Optimal Transport

• Host: GitHub
• URL: https://github.com/pythonot/coot
• Owner: PythonOT
• License: mit
• Created: 2020-10-21T14:44:32.000Z (over 4 years ago)
• Default Branch: main
• Last Pushed: 2020-11-06T13:09:58.000Z (over 4 years ago)
• Last Synced: 2025-03-25T08:11:35.299Z (about 2 months ago)
• Language: Jupyter Notebook
• Size: 42.8 MB
• Stars: 42
• Watchers: 3
• Forks: 12
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# COOT

Python3 implementation of the paper [CO-Optimal Transport](https://arxiv.org/abs/2002.03731) (NeurIPS 2020)

CO-Optimal Transport (COOT) is an Optimal Transport problem between measures whose supports do not necessarily live in the same metric space. It aims to simultaneously optimize two transport maps between both samples and features of two datasets with possibly different number of rows and columns.

Feel free to ask if any question.

If you use this toolbox in your research and find it useful, please cite COOT using the following bibtex reference:

```

@incollection{coot_2020,

title = {CO-Optimal Transport},

author = {Redko, Ievgen, Vayer, Titouan, Flamary, R\'{e}mi and Courty, Nicolas},

booktitle = {Advances in Neural Information Processing Systems 33},

year = {2020}

}

```

### Prerequisites

* Numpy (>= 1.11)

* Matplotlib (>= 1.5)

* For Optimal transport [Python Optimal Transport](https://pythonot.github.io/) POT (>=0.5.1)

### What is included ?

* The main function that computes the COOT problem between two datasets in the code folder (file cot.py)

* Demo notebooks:

	- [coot_mnist.ipynb](./example/coot_mnist.ipynb): COOT on the MNIST/USPS problem

![](./example/mnist_usps.png)

* Code to repoduce results of the paper in the expe folder.

![](./expe/clootclustering_faces.png)

### Authors

* [Ievgen Redko](https://ievred.github.io/)

* [Titouan Vayer](https://github.com/tvayer)

* [Rémi Flamary](https://github.com/rflamary)

* [Nicolas Courty](https://github.com/ncourty)