Pranshu-Bahadur / kmeans-pytorch Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Here's kmeans in pytorch with multi-gpu support, for all your unsupervised learning needs!

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 13 Commits
__pycache__		__pycache__
LICENSE		LICENSE
README.md		README.md
kmeans\|\|-2020-Paper_Summary.pdf		kmeans\|\|-2020-Paper_Summary.pdf
notes		notes
pytorch_kmeans.py		pytorch_kmeans.py
test.py		test.py

Repository files navigation

Kmeans approximation algorithms for GPU usage with pytorch.

Note: This implementation supports multi-GPUs!!!

Dependencies:

   pip install torch

Current State:

Implementing center initalizers:

naive kmeans (uniform distribution)
kmeans++ (D^2 distribution)
kmeans-|| Oversampling

Implementations:

Lloyd's Algorithm (a.k.a Naive Kmeans)
kmeans++
kmeans-||

About

Here's kmeans in pytorch with multi-gpu support, for all your unsupervised learning needs!

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%