python (12.9k questions)

javascript (9.2k questions)

reactjs (4.7k questions)

java (4.2k questions)

java (4.2k questions)

c# (3.5k questions)

c# (3.5k questions)

html (3.3k questions)

Questions - silhouette

Is there any cluster evaluation method implemented in pyTorch?

I have found k-means implementation in PyTorch using GPUs which is 30 times faster than CPUs. Is there any method (such as Silhouette score, Dunn's index, ...) implemented preferably in PyTorch that ...
test-img

Mehrdad Mirzaei

pytorch

gpu

cluster-analysis

evaluation

silhouette

pytorch

gpu

cluster-analysis

evaluation

silhouette

Votes: 0

Answers: 1

Latest Answer

I implemented the silhouette score in PyTorch based on a numpy implementation from Alexandre Abraham (https://gist.github.com/AlexandreAbraham/5544803): https://github.com/maxschelski/pytorch-cluster-...
test-img

Max Schelski

Posts

Questions

Blogs

Jobs