Icold21/FastKMeans
A highly scalable, GPU-accelerated multi-prototype Soft K-Means classifier built on PyTorch. Engineered for massive datasets, extreme sparsity (e.g., TF-IDF NLP embeddings), and tens of thousands of classes. Features streaming/online learning, automatic centroid pruning, and mixed-precision Tensor Core support.
Stars
—
Forks
—
Language
Jupyter Notebook
License
MIT
Last pushed
Apr 09, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/Icold21/FastKMeans"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
scikit-learn/scikit-learn
scikit-learn: machine learning in Python
pallets/click
Python composable command line interface toolkit
Farama-Foundation/Gymnasium
An API standard for single-agent reinforcement learning environments, with popular reference...
probabl-ai/skore
Track your Data Science. Skore's open-source Python library accelerates ML model development...
huggingface/evaluate
🤗 Evaluate: A library for easily evaluating machine learning models and datasets.