Skip to content

brendanmallery9/Entropic-Barycenters

Repository files navigation

This file contains the code required to reproduce results in main submission.

Requirements:

numpy
torch
matplotlib 
sklearn (sci-kit learn)
pandas
Python optimal transport (https://pythonot.github.io/index.html)
h5py
scipy ()
tensorflow

For numerical experiments for the analysis problem: 

1d_gauss.py
box_experiment.py
high_dim_gaussian_analysis_experiment.py

For classification: 
classification_example.py

For completion:
pointcloud_completion.py

(*) Set working directory to '/classification_experiment/' before running classification_example.py

Pointnet implementation is adapted from:

https://github.com/nikitakaraevv/pointnet (original paper: https://arxiv.org/abs/1612.00593)

as detailed in main submission.


Score estimation implementation is adapted from:

https://github.com/miskcoo/kscore (original paper: https://arxiv.org/abs/2005.10099)

as detailed in main submission.


We have included a subset of the PointCloud-c dataset (classificaton_exmaple/pointcloud-c). Full dataset is available at https://pointcloud-c.github.io/home.html

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages