1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
//! Probabilistic clustering algorithms using Bayesian nonparametric models.
//!
//! This module provides three complementary probabilistic clustering algorithms:
//!
//! * [`dpgmm`] – Dirichlet Process Gaussian Mixture Model (DP-GMM) via
//! variational Bayes inference with stick-breaking representation.
//!
//! * [`crp`] – Chinese Restaurant Process (CRP) sampler and Gibbs-sampling
//! based CRP mixture inference, plus the generalized Pitman-Yor process.
//!
//! * [`variational_gmm`] – Full Variational Bayesian GMM with a fixed number
//! of components, component pruning, and variational BIC model selection.
//!
//! # Overview
//!
//! Bayesian nonparametric mixture models allow the data itself to determine
//! the effective number of clusters. The Dirichlet Process is the canonical
//! prior over infinite mixture models: with concentration `α` it generates
//! on the order of `α log N` clusters for N observations.
//!
//! ## Example – DP-GMM
//!
//! ```rust
//! use scirs2_cluster::probabilistic::dpgmm::{DPGMMConfig, DPGMMModel};
//! use scirs2_core::ndarray::Array2;
//!
//! let data = Array2::from_shape_vec((8, 2), vec![
//! 0.0, 0.0, 0.1, 0.1, -0.1, 0.0, 0.0, -0.1,
//! 5.0, 5.0, 5.1, 4.9, 4.9, 5.1, 5.0, 5.0,
//! ]).expect("data");
//!
//! let cfg = DPGMMConfig::new(1.0, 8, 100);
//! let model = DPGMMModel::new(cfg);
//! let result = model.fit(data.view()).expect("fit");
//! let labels = result.predict(data.view()).expect("predict");
//! assert_eq!(labels.len(), 8);
//! ```
//!
//! ## Example – CRP
//!
//! ```rust
//! use scirs2_cluster::probabilistic::crp::CRPSampler;
//!
//! let sampler = CRPSampler::new(2.0);
//! let assignments = sampler.sample_seating(20).expect("sample");
//! assert_eq!(assignments.len(), 20);
//! ```
//!
//! ## Example – Variational GMM
//!
//! ```rust
//! use scirs2_cluster::probabilistic::variational_gmm::{VBGMMConfig, VBGMMModel};
//! use scirs2_core::ndarray::Array2;
//!
//! let data = Array2::from_shape_vec((8, 2), vec![
//! 0.0, 0.0, 0.1, 0.1, -0.1, 0.0, 0.0, -0.1,
//! 5.0, 5.0, 5.1, 4.9, 4.9, 5.1, 5.0, 5.0,
//! ]).expect("data");
//!
//! let cfg = VBGMMConfig::new(4, 100, 1e-5);
//! let model = VBGMMModel::new(cfg);
//! let result = model.fit_vbem(data.view()).expect("fit");
//! let labels = result.predict(data.view()).expect("predict");
//! assert_eq!(labels.len(), 8);
//! ```
// Re-export commonly used types for convenience.
pub use ;
pub use ;
pub use ;