Skip to content

smiler96/GMM-KMeans-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Implementation of the batch GMM & k-means with PyTorch

To cluster large amounts of data with K-Means, accelerated by GPU.

  • Data: 500000x3 K=5

  • Data: 50000000x3 K=5

To cluster large amounts of data with Gaussian Mixture Model, accelerated by GPU.

  • Data: 100000x512 K=5

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages