All Projects → ilyaraz → pytorch_kmeans

ilyaraz / pytorch_kmeans

Licence: MIT License
Implementation of the k-means algorithm in PyTorch that works for large datasets

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to pytorch kmeans

Hazelcast Cpp Client
Hazelcast IMDG C++ Client
Stars: ✭ 67 (+76.32%)
Mutual labels:  big-data, clustering
Hazelcast Nodejs Client
Hazelcast IMDG Node.js Client
Stars: ✭ 124 (+226.32%)
Mutual labels:  big-data, clustering
Moosefs
MooseFS – Open Source, Petabyte, Fault-Tolerant, Highly Performing, Scalable Network Distributed File System (Software-Defined Storage)
Stars: ✭ 1,025 (+2597.37%)
Mutual labels:  big-data, clustering
Hazelcast Go Client
Hazelcast IMDG Go Client
Stars: ✭ 140 (+268.42%)
Mutual labels:  big-data, clustering
R-stats-machine-learning
Misc Statistics and Machine Learning codes in R
Stars: ✭ 33 (-13.16%)
Mutual labels:  clustering, k-means
scarf
Toolkit for highly memory efficient analysis of single-cell RNA-Seq, scATAC-Seq and CITE-Seq data. Analyze atlas scale datasets with millions of cells on laptop.
Stars: ✭ 54 (+42.11%)
Mutual labels:  big-data, clustering
Hazelcast Python Client
Hazelcast IMDG Python Client
Stars: ✭ 92 (+142.11%)
Mutual labels:  big-data, clustering
Clustering4Ever
C4E, a JVM friendly library written in Scala for both local and distributed (Spark) Clustering.
Stars: ✭ 126 (+231.58%)
Mutual labels:  big-data, clustering
A-quantum-inspired-genetic-algorithm-for-k-means-clustering
Implementation of a Quantum inspired genetic algorithm proposed by A quantum-inspired genetic algorithm for k-means clustering paper.
Stars: ✭ 28 (-26.32%)
Mutual labels:  clustering, k-means
hazelcast-csharp-client
Hazelcast .NET Client
Stars: ✭ 98 (+157.89%)
Mutual labels:  big-data, clustering
ClusterAnalysis.jl
Cluster Algorithms from Scratch with Julia Lang. (K-Means and DBSCAN)
Stars: ✭ 22 (-42.11%)
Mutual labels:  clustering, k-means
genie
Genie: A Fast and Robust Hierarchical Clustering Algorithm (this R package has now been superseded by genieclust)
Stars: ✭ 21 (-44.74%)
Mutual labels:  clustering
watchman
Watchman: An open-source social-media event-detection system
Stars: ✭ 18 (-52.63%)
Mutual labels:  clustering
v6.dooring.public
可视化大屏解决方案, 提供一套可视化编辑引擎, 助力个人或企业轻松定制自己的可视化大屏应用.
Stars: ✭ 323 (+750%)
Mutual labels:  big-data
falcon
Mirror of Apache Falcon
Stars: ✭ 95 (+150%)
Mutual labels:  big-data
Fred
A fast, scalable and light-weight C++ Fréchet distance library, exposed to python and focused on (k,l)-clustering of polygonal curves.
Stars: ✭ 13 (-65.79%)
Mutual labels:  clustering
machine-learning-course
Machine Learning Course @ Santa Clara University
Stars: ✭ 17 (-55.26%)
Mutual labels:  clustering
Spectre
A computational toolkit in R for the integration, exploration, and analysis of high-dimensional single-cell cytometry and imaging data.
Stars: ✭ 31 (-18.42%)
Mutual labels:  clustering
tf-example-models
TensorFlow-based implementation of (Gaussian) Mixture Model and some other examples.
Stars: ✭ 42 (+10.53%)
Mutual labels:  k-means
DBSCANSD
Java implementation for DBSCANSD, a trajectory clustering algorithm.
Stars: ✭ 35 (-7.89%)
Mutual labels:  clustering

PyTorch implementation of the k-means algorithm

This code works for a dataset, as soon as it fits on the GPU. Tested for Python3 and PyTorch 1.0.0.

For simplicity, the clustering procedure stops when the clustering stops updating. In practice, this might be too strict and should be relaxed.

There is a magic constant (search for chunk_size) which should ideally be determined automatically based on the amount of free memory on the GPU.

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].