deepbox/ml
Clustering
Partitioning, density-based, graph-based, hierarchical, and mixture clustering algorithms.
Unsupervised
AffinityPropagation
Affinity Propagation clustering algorithm.
AgglomerativeClustering
Agglomerative (hierarchical) clustering.
Birch
BIRCH clustering algorithm.
DBSCAN
DBSCAN (Density-Based Spatial Clustering of Applications with Noise).
GaussianMixture
Gaussian Mixture Model (GMM).
KMeans
K-Means clustering algorithm.
MeanShift
Mean Shift clustering using a flat/Gaussian kernel.
MiniBatchKMeans
Mini-Batch K-Means clustering.
OPTICS
OPTICS clustering algorithm.
SpectralClustering
Spectral Clustering using graph Laplacian eigenvectors.
ml-clustering.ts
import { AgglomerativeClustering, DBSCAN, GaussianMixture, KMeans, SpectralClustering,} from "deepbox/ml";import { tensor } from "deepbox/ndarray";const X = tensor([[1, 1], [1.2, 0.8], [4, 4], [4.2, 4.1]]);console.log(new KMeans({ nClusters: 2 }).fit(X).predict(X).toString());console.log(new DBSCAN({ eps: 0.6 }).fit(X).labels.toString());console.log(new AgglomerativeClustering({ nClusters: 2 }).fit(X).labels.toString());console.log(new GaussianMixture({ nComponents: 2 }).fit(X).predict(X).toString());console.log(new SpectralClustering({ nClusters: 2 }).fit(X).labels.toString());