Code to reproduce experiments of the ICML 2022 Paper Distribution Regression with Sliced Wasserstein Kernels

ArXiv link: https://arxiv.org/abs/2202.03926. Proceeding link: https://proceedings.mlr.press/v162/meunier22b.html.

Installation

Using conda: clone the repository, and once inside it, execute the following bash commands

CPU install

conda create -n drswk && conda activate drswk

# To run experiments
conda install pytorch scikit-learn

# To plot experiment results and use the notebooks
conda install ipympl pandas 

Cite The paper:

@InProceedings{meunier2022dregr,
  title = 	 {Distribution Regression with Sliced {W}asserstein Kernels},
  author =       {Meunier, Dimitri and Pontil, Massimiliano and Ciliberto, Carlo},
  booktitle = 	 {Proceedings of the 39th International Conference on Machine Learning},
  pages = 	 {15501--15523},
  year = 	 {2022},
}