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[source]

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