Search Packages

kmeans1d

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

neurocaps

Co-activation Patterns (CAPs) Python package

gap-stat

Python implementation of the gap statistic with optional Rust optimizations.

clumppy

Clustering algorithms

pycave

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

pdc-dp-means

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

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

colorz

Color scheme generator.

libkmcuda

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

metacluster

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

rectokens

A tokenizer library for sequential recommendation systems

raster2poly

Classify rasters (KMeans, Random Forest, DN rules) and vectorise to polygons.

allmeans

Automatic topic modelling using minimal external input and computational resources

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.

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)

vlac

Vectors of Locally Aggregated Concepts

pytrain

Machinelearning library for python

pek

Progressive Ensemble K-means clustering

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

A python library for quantum machine learning and quantum deep learning, built on top of qiskit and pennylane

vectors-of-locally-aggregate-concepts

Vectors of Locally Aggregated Concepts

artutils

A modular ML-powered palette toolkit for artists and designers

hgmeans

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