Finding optimal k in k-means using bayesian optimization
Project description
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()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
bayesKMeans-0.3.4.tar.gz
(5.6 kB
view hashes)