src.spectral_clustering.kmeans_predict

src.spectral_clustering.kmeans_predict(X: torch.Tensor, cluster_centers: torch.Tensor, distance: str = 'euclidean', device=device(type='cpu')) → torch.Tensor

Kmeans prediction using existing cluster centers.

Parameters

• X – matrix

• cluster_centers – cluster centers

• distance – distance [options: ‘euclidean’, ‘cosine’] [default: ‘euclidean’]

• device – computing device [default: ‘cpu’]

Returns

cluster ids