Clustering Python Packages

mteb

MTEB: Massive Text Embedding Benchmark

hdbscan

A high performance implementation of HDBSCAN clustering.

drain3

A robust streaming log template miner based on the Drain algorithm

usearch

Fast Open-Source Search & Clustering engine × for Vectors & Arbitrary Objects × in C++, C, Python, JavaScript, Rust, Java, Objective-C, Swift, C#, GoLang, and Wolfram 🔍

pycaret

Open-source, low-code AutoML platform for Python. PyCaret 4.0: sklearn-native engine + React control plane.

dtaidistance

Time series distances: Dynamic Time Warping (fast DTW implementation in C)

pylibraft-cu12

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

libraft-cu12

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

dtw-python

Python port of R's Comprehensive Dynamic Time Warp algorithms package

k-means-constrained

K-Means clustering - constrained with minimum and maximum cluster size. Documentation: https://joshlk.github.io/k-means-constrained

hazelcast-python-client

Hazelcast Python Client

pypots

A Python toolkit/library for reality-centric machine/deep learning & data mining on partially-observed time series, with 50+ SOTA neural network models for scientific analysis tasks (imputation, classification, clustering, forecasting, anomaly detection, cleaning) on incomplete industrial irregularly-sampled multivariate TS with NaN missing values

raft-dask-cu12

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.

orange3

🍊 :bar_chart: :bulb: Orange: Interactive data analysis

dedupe

:id: A python library for accurate and scalable fuzzy matching, record deduplication and entity-resolution.

nomic

Nomic Developer API SDK

torch-kmeans

PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.

aimsim-core

A Python toolbox to work with molecular similarity

minisom

:red_circle: MiniSom is a minimalistic implementation of the Self Organizing Maps

libcuvs-cu12

cuVS - a library for vector search and clustering on the GPU

scikit-fda

Functional Data Analysis Python package

banditpam

BanditPAM C++ implementation and Python package

dbscan

Theoretically Efficient and Practical Parallel DBSCAN

cuvs-cu12

cuVS - a library for vector search and clustering on the GPU