Contrastive-Clustering
Contrastive-Clustering copied to clipboard
推理时候使用k-means
hi~yunfan: 请教一个问题, 训练时使用instance-loss 和cluster-loss,在推理的时候如果直接采用k-means的效果会比使用本文的cluster-head进行推理的效果相差很大吗?
您好,这个结果可以参照我们的期刊扩展https://github.com/Yunfan-Li/Twin-Contrastive-Learning ,效果相差不大,不过cluster head稍好一些