Inquiry about the centroids of the K-means method

[hellodrx]

Hi, firstly thanks for your wonderful work.

I want to get the centroids of the clusters and visualize them. However, from your introduction, it seems I can only get the labels of all samples. Do you have any suggestions that I can get the results?

Thanks again for helping me out.

[hellodrx]

Is this the function I can use to get the centroids

TorchPQ/torchpq/clustering/KMeans.py

Line 365 in b8bbadf

def compute_centroids(self, data, labels):

[DeMoriarty]

Hi, centroids are stored in instance variable named"centroids".

kmeans = KMeans(n_clusters=4)
kmeans.fit(x)
centroids = kmeans.centroids

[hellodrx]

Hi, centroids are stored in instance variable named"centroids".

kmeans = KMeans(n_clusters=4)
kmeans.fit(x)
centroids = kmeans.centroids

Thank you very much!