Metadata-Version: 2.1
Name: bayesKMeans
Version: 0.3.5
Summary: Finding optimal k in k-means using bayesian optimization
Home-page: https://github.com/ForseIKomar/BayesKMeans
Author: Aleksandr Burlakov
Author-email: samitist11@gmail.com
License-File: LICENSE

Example of work:

from bayeskmeans.bayes_kmeans import BayesKMeans
from bayeskmeans.bayes_visualize import BayesKMeansVisualize
from sklearn.datasets import make_blobs

data = make_blobs(n_samples=1000, n_features=2, centers=21, cluster_std=5, center_box=(-300, 300))
data = data[0]

bayesKMeans = BayesKMeans(data)

bayesKMeans.find_k()

print(bayesKMeans.found_k)

visual = BayesKMeansVisualize(bayesKMeans)
visual.show_bayesian_plot()
