Implementation of Nested Mini-Batch K-Means in numpy - Newling et al.
- Import the class
from NestedKMeans import NestedKMeans
- Initialize the model
nkmeans = NestedKMeans(<datapoints>, <No. of Clusters>, <No. of starting Points>>, rho=<>, earlyStop=<To stop early assign True>)
- Train
nkmeans.train(<No. of max iters>)
- Show the trained result
# works only for 2 dimensions
nkmeans.show()