def test_basic(self, single_chunk_blobs): X, y = single_chunk_blobs a = cluster.PartialMiniBatchKMeans(n_clusters=3, random_state=0) b = cluster_.MiniBatchKMeans(n_clusters=3, random_state=0) a.fit(X) b.partial_fit(X) assert_estimator_equal(a, b, exclude=['random_state_'])
from dask_ml import cluster import numpy as np if __name__ == '__main__': X = np.random.rand(10000, 16) kmeans = cluster.PartialMiniBatchKMeans(n_clusters=500, init='k-means++').partial_fit(X) print(kmeans.labels_)