Kmeans Python Packages

kmeans1d

A Python package for optimal 1D k-means clustering.

neurocaps

A Python package for performing Co-Activation Patterns (CAPs) analyses on resting-state and task-based fMRI data.

clumppy

Clustering algorithms

pycave

Traditional Machine Learning Models for Large-Scale Datasets in PyTorch.

cake-ensemble

Confidence in Assignments via K-partition Ensembles

gap-stat

Dynamically get the suggested clusters in the data for unsupervised learning.

colorz

Color scheme generator.

allmeans

Automatic topic discovery with zero LLMs, minimal input

metacluster

MetaCluster: An Open-Source Python Library for Metaheuristic-based Clustering Problems

superkmeans

⚡ Super fast clustering for high-dimensional vectors on CPUs (x86, ARM) and GPUs — for Python and C++. 100x faster clustering of vector embeddings than FAISS

pdc-dp-means

"Revisiting DP-Means: Fast Scalable Algorithms via Parallelism and Delayed Cluster Creation" [Dinari and Freifeld, UAI 2022]

constrainedkmeans

Constrained KMeans algorithm.

optimalcluster

OptimalCluster is the Python implementation of various algorithms to find the optimal number of clusters. The algorithms include elbow, elbow-k_factor, silhouette, gap statistics, gap statistics with standard error, and gap statistics without log. Various types of visualizations are also supported.

libkmcuda

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA

vectors-of-locally-aggregate-concepts

Vectors of Locally Aggregated Concepts

vlac

Tool for creating document features

hgpack

Source code of HG-means clustering, from "HG-means: A scalable hybrid genetic algorithm for minimum sum-of-squares clustering". (Gribel and Vidal, 2019)

pytrain

Machine Learning library for python

flash-kmeans-mlx

IO-aware batched K-Means for Apple Silicon, ported from Flash-KMeans (Triton/CUDA) to pure MLX. Up to 94x faster than sklearn.

qlearnkit

No description available

clus

Clustering and fuzzy-clustering library with integrated CLI

pek

PEK - Progressive Ensemble K-means clustering

artutils

Image-to-art tools: palettes, gradients, opposites - in Python

rectokens

[Pytorch] Efficient tokenization library for recommendations and generative retrieval using pre-trained language models. Inspired by "Vectorizing the Trie: Efficient Constrained Decoding for LLM-based Generative Retrieval on Accelerators"