No project description provided
Project description
K Means using PyTorch
PyTorch implementation of kmeans for utilizing GPU
Getting Started
import torch
import numpy as np
from kmeans_pytorch import kmeans
# data
data_size, dims, num_clusters = 1000, 2, 3
x = np.random.randn(data_size, dims) / 6
x = torch.from_numpy(x)
# kmeans
cluster_ids_x, cluster_centers = kmeans(
X=x, num_clusters=num_clusters, distance='euclidean', device=torch.device('cuda:0')
)
see example.ipynb
for a more elaborate example
Requirements
- PyTorch version >= 1.0.0
- Python version >= 3.6
Installation
install with pip
:
pip install kmeans-pytorch
Installing from source
To install from source and develop locally:
git clone https://github.com/subhadarship/kmeans_pytorch
cd kmeans_pytorch
pip install --editable .
CPU vs GPU
see cpu_vs_gpu.ipynb
for comparison between CPU and GPU
Notes
- useful when clustering large number of samples
- utilizes GPU for faster matrix computations
- support euclidean and cosine distances (for now)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
kmeans_pytorch-0.3.tar.gz
(4.3 kB
view hashes)
Built Distribution
Close
Hashes for kmeans_pytorch-0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7c0ddf1c19baaea83cd5494c3fdb86f79c0ba11c32848ba542aa190350f7c9eb |
|
MD5 | d551470eeee439cdea99a881289d2c04 |
|
BLAKE2b-256 | b5c9eb5b82e7e9741e61acf1aff70530a08810aa0c7e2272c534ff7a150fc5bd |