What problem does it solve?
UMAP-Learn solves the problem of producing low-dimensional embeddings from high-dimensional data so you can visualize structure and prepare features for downstream learning.
Core Features & Use Cases
- Fast nonlinear dimensionality reduction: Generate 2D/3D embeddings that preserve local and broader manifold structure for exploratory analysis.
- Scalable embedding for ML workflows: Use higher-dimensional embeddings (e.g., 5–10+) as preprocessing or feature engineering for clustering and classifiers.
- Supervised and semi-supervised embedding: Incorporate labels via y to improve class separation, including partial-label scenarios.
- Clustering preprocessing with HDBSCAN-ready embeddings: Improve density-based clustering by reducing dimensionality while keeping neighborhood relationships.
- Practical extensions: Support transform() for projecting new data, plus parametric UMAP for efficient learned mappings when you need to reuse the embedding function.
Quick Start
Standardize your features first and then fit a UMAP reducer to produce a 2D embedding suitable for immediate plotting, clustering, or sanity checks.