Multilevel clustering with Wasserstein means and its variants

Code to reproduce the results in the paper 'On Efficient Multilevel Clustering via Wasserstein Distances', JMLR 2021.

Tested with Python 3.6.

Citation

Viet Huynh, Nhat Ho, Nhan Dam, XuanLong Nguyen, XuanLong Nguyen, Hung Bui, and Dinh Phung, 'On Efficient Multilevel Clustering via Wasserstein Distances', Journal of Machine Learning Research, 2021.

Bibtex

@article{huynh_etal_jmlr_onfficient,
  author    = {Viet Huynh, Nhat Ho, Nhan Dam, XuanLong Nguyen, XuanLong Nguyen, Hung Bui, and Dinh Phung},
  title     = {On Efficient Multilevel Clustering via Wasserstein Distances},
  journal = {Journal of Machine Learning Research},
  year      = {2021} 
}