kasba_py
Source: src/kasba_py.rs
PyO3 bindings for KASBA clustering (issue #192).
Exposes kasba_fit and kasba_predict as Python-callable functions
that accept 3D numpy arrays of shape (n_cases, n_channels, n_timepoints).
Python API
kasba_fit
Fit KASBA clustering on multivariate time series data.
Arguments
| Parameter | Description |
|---|---|
data |
3D array of shape (n_cases, n_channels, n_timepoints), f64 |
n_clusters |
Number of clusters |
c |
MSM cost parameter |
independent |
MSM mode (true=per-channel, false=multivariate dependent) |
ba_subset_size |
Fraction of cluster members for barycenter averaging |
initial_step_size |
Starting SGD step size |
decay_rate |
Exponential decay rate for step size |
n_iters |
Maximum clustering iterations |
random_seed |
Random seed for determinism |
Returns
Tuple of (labels, centers, inertia, n_iter)
kasba_predict
Predict cluster assignments for new data using pre-computed centers.
Arguments
| Parameter | Description |
|---|---|
centers |
2D array of shape (n_clusters, n_channels * n_timepoints) |
data |
3D array of shape (n_cases, n_channels, n_timepoints) |
c |
MSM cost parameter |
independent |
MSM mode |
Returns
1D array of cluster labels