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 .
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.2.tar.gz
(4.1 kB
view hashes)
Built Distribution
Close
Hashes for kmeans_pytorch-0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 836962e04bdecefb7086dc88934c2e243d1588f37397e2852dc3c0c7d309a16b |
|
MD5 | f0b2f7022b6f1ce48acfbc06076f1536 |
|
BLAKE2b-256 | 8c894b91f8c90216d3f724b11cde784467f886feb2d7823a563c9961c78c0217 |