The repo contains the source code needed to reproduce the results in Clustering with UMAP: Why and How Connectivity Matters by Dalmia and Sia
The code for this paper is all implemented in Colab/jupyter notebooks for ease of reading and understanding. All the required modules are included in the firsat section of the notebook
In the colab notebook, we walk you through our modification to the orginial UMAP algorithm (for more specifics on how the UMAP algorithm works and implemented, please refer to the orginial UMAP repository (v0.5+)
There are two different colab notebooks provided in the repository:
Both implementations contain all the connectivity method (NN<MST-min<MST-all) and can e easily modified to include others