Skip to content
/ JDOT Public

Joint distribution optimal transportation for domain adaptation

License

Notifications You must be signed in to change notification settings

rflamary/JDOT

Repository files navigation

JDOT

Joint distribution optimal transportation for domain adaptation

imgreg

This repository contains the code used for the numerical experiments of the foillowing paper:

N. Courty, R. Flamary, A. Habrard, A. Rakotomamonjy, "Joint Distribution Optimal Transportation for Domain Adaptation", Neural Information Processing Systems (NIPS), 2017.

The code is under MIT Licence but please refer to and cite the above paper if you use it for academic purposes.

Note that this code has not been thouroughly tested and a more clean/robust implementation will be added to the POT Toolbox that already have numerous optimal transport domain adaptation methods.

Dependencies

In order to run, the code requires the following Python modules:

  • Numpy/Scipy
  • Matplotlib
  • POT

If you want to use the neural network JDOT example you will also need the keras toolbox.

If you have not already installed them you can install the dependencies with PIP using the following command

$ pip install numpy scipy matplotlib POT # keras

Modules

  • jdot: module containing the thre implemented variants of JDOT (Kernel Ridge Regression, Hinge Loss, least square neural network).
  • classif: simple classifier classes used in JDOT.

Plot examples

We provide 3 example scripts that reproduce the Figures in the paper.

  • visu_regression.py Plot Figure 1 in the paper (regression toy example, see below).

imgreg

  • visu_classification.py Plot Figure 1 in the supplementary (classification toy example with BCD iterations).

imgreg

  • visu_regression_nn.py shows an exmaple of JDOT when learning a keras neural network.

About

Joint distribution optimal transportation for domain adaptation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages