search

caponetto / bayesian-hierarchical-clustering

Python implementation of Bayesian hierarchical clustering and Bayesian rose trees algorithms.
GNU General Public License v3.0
6 stars 2 forks source link
bayesian-cluster-analysis bayesian-statistics clustering hierarchical-clustering python

readme

CI License: GPL v3

Bayesian Hierarchical Clustering in python

This is a python implementation of the Bayesian Hierarchical Clustering algorithm proposed by Heller & Ghahramani (2005).

HELLER, Katherine A.; GHAHRAMANI, Zoubin. Bayesian hierarchical clustering. In: Proceedings of the 22nd international conference on Machine learning. 2005. p. 297-304.

It also includes the Bayesian Rose Trees extension proposed by Blundell et al (2012).

BLUNDELL, Charles; TEH, Yee Whye; HELLER, Katherine A. Bayesian rose trees. arXiv preprint arXiv:1203.3468, 2012.

How to build from source

  1. Create an anaconda environment using the environment.yml file.

    $ conda env create -f environment.yml

  2. Activate the environment after the installation is completed.

    $ conda activate bayesian-hierarchical-clustering

  3. Install the following libraries: pytest, coverage and flake8.

    $ conda install pytest coverage flake8

  4. Run the build_dev.sh script.

    $ ./build_dev.sh

The files will be available in the dist folder.

Examples

Examples can be found at caponetto/bayesian-hierarchical-clustering-examples.

Contribute

All contributions are welcome, so don't hesitate to submit a pull request. ;-)

License

This code is released under GPL 3.0 License.

Check LICENSE file for more information.