KMeans is not implemented in the original kmeans_pytorch

facebookresearch / MetaCLIP

ICLR2024 Spotlight: curation/training code, metadata, distribution and pre-trained models for MetaCLIP; CVPR 2024: MoDE: CLIP Data Experts via Clustering

Other

1.19k stars 50 forks source link

KMeans is not implemented in the original kmeans_pytorch #54

Open rezoo opened 4 months ago

rezoo commented 4 months ago

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 commented 4 months ago

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.