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

Using Conda (Recommended)

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. Run the build script.

   $ ./build_dev.sh

Using pip

Alternatively, if you prefer using pip:

1. Create a virtual environment.

   $ python -m venv venv
   $ source venv/bin/activate  # On Windows: venv\Scripts\activate

2. Install development dependencies.

   $ pip install -r requirements-dev.txt

3. Run the build script.

   $ ./build_dev.sh

The built packages 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.