/k_medoids

PyTorch Implementation of k-medoids

Primary LanguageJupyter Notebook

k_medoids

This repository is a PyTorch implementation of the k-medoids algorithm from the paper "A Simple and Fast Algorithm for K-medoids Clustering".

The example in the jupyter notebook tries to find 3 medoids from a Gaussian mixture model of 3 clusters.

before_k_medoids after_k_medoids