Nested Mini-Batch K-Means

Implementation of Nested Mini-Batch K-Means in numpy - Newling et al.

arxiv 1602.02934

Usage

1. Import the class

from NestedKMeans import NestedKMeans

2. Initialize the model

nkmeans = NestedKMeans(<datapoints>, <No. of Clusters>, <No. of starting Points>>, rho=<>, earlyStop=<To stop early assign True>)

3. Train

nkmeans.train(<No. of max iters>)

4. Show the trained result

# works only for 2 dimensions
nkmeans.show()