jwyang/JULE.torch

How to get the clustered data after training?

Paper99 opened this issue · 3 comments

Paper99 commented

Many Thanks for your code.
And I still have a question that HOW TO GET THE CLUSTERED DATA AFTER TRAINING.

jwyang commented

Hi, You will get the predicted label at the ending, and also learned representation:

Line 454-456 for saving predicted labels, and line 299-301 for saving learned representations.

  • 👍3
Paper99 commented

Thank you a lot.

VeLKerr commented

Hi, @jwyang, thanks a lot for the explanation above.

I've retrieved labels of each 20 epochs and feature matrices of last (100th) epoch. But I need to find out the cluster of each image. How I can correlate resulting labels and input images? Can we be sure that labels[0] corresponds to 0_cat.png (the 1st image in CIFAR-10)?

Thank you in advance for the answer.