search

deu30303 / RUC

144 stars 15 forks source link

How to get my clustering result? #12

Closed zbw0329 closed 3 years ago

zbw0329 commented 3 years ago

Hi~Thanks for your excellent work. I trained RUC on CIFAR10 datasets,but I just get the trained model. I want to know how to get clustering result? If I want to save label corresponding to each sample to a txt file , what should I do?

deu30303 commented 3 years ago

Hi, thank you for your interest in our work!

We evaluate our model based on hungarian_match algorithm. (See lib/protocols.py _hungarian_match function) You can get the clustering result by matching the label index.

Thank you again for your interest.