KMeans

KMeans clustering.

Parameters

• documents_embeddings (torch.Tensor)

• n_clusters (int)

• max_iter (int)

• n_init (int)

• seed (int)

• device (str)