KMeans is not implemented in the original kmeans_pytorch

[rezoo]

Currently I am trying to reproduce MoDE in my own dataset.
Although it worked successfully up to feature extraction, problems arose with kmeans clustering.
Your repository describes that mode/prep_hrchy.py imports KMeans class from kmeans_pytorch, however, the original kmeans_pytorch does not support such a class.
I think that this library may be a modified one, but is this implementation published anywhere?

[Phoenix-V]

Thank you for your question. Regarding the KMeans installation, please refer the following command. The corresponding ReadMe will be updated shortly.

git clone https://github.com/kernelmachine/balanced-kmeans.git
cd balanced-kmeans
pip3 install -e .

More details of KMeans usage can be found here.