Function kmeans_example

examples/kmeans.py:7–17

(plot=False)

def kmeans_example(plot=False):
    X, y = make_blobs(
        centers=4, n_samples=500, n_features=2, shuffle=True, random_state=42
    )
    clusters = len(np.unique(y))
    k = KMeans(K=clusters, max_iters=150, init="++")
    k.fit(X)
    k.predict()

    if plot:
        k.plot()


if __name__ == "__main__":