search

subhadarship / kmeans_pytorch

kmeans using PyTorch
https://subhadarship.github.io/kmeans_pytorch
MIT License
479 stars 77 forks source link

Clustering of batch data #29

Open mliwang opened 3 years ago

mliwang commented 3 years ago

Does this method support the clustering of batch data, that is, the data that is sent in (batch-size,n,dim), and does it support the clustering of n sample points with dim dimension in each batch?

laoyangui commented 3 years ago

@mliwang here is another implementation of k-means with minibatch in PyTorch.