models module

mvcluster.models.update_rule_F(XW: Tensor, G: Tensor, k: int) → Tensor

Update centroids by mean of embeddings per cluster.

Parameters:

• XW – Tensor [n_samples, emb_dim], embeddings.

• G – Tensor [n_samples], cluster assignments.

• k – Number of clusters.

mvcluster.models.update_rule_G(XW: Tensor, F: Tensor) → Tensor

Assign samples to nearest centroid.

Parameters:

• XW – Tensor [n_samples, emb_dim], embeddings.

• F – Tensor [k, emb_dim], centroids.

mvcluster.models.update_rule_W(X: Tensor, F: Tensor, G: Tensor) → Tensor

Update projection matrix via orthogonal Procrustes.

Parameters:

• X – Tensor [n_samples, n_features].

• F – Tensor [n_samples, emb_dim], centroids.

• G – Tensor [n_samples], cluster assignments.