Contrastive-Clustering icon indicating copy to clipboard operation
Contrastive-Clustering copied to clipboard

推理时候使用k-means

Open topDreamer opened this issue 1 year ago • 1 comments

hi~yunfan: 请教一个问题, 训练时使用instance-loss 和cluster-loss,在推理的时候如果直接采用k-means的效果会比使用本文的cluster-head进行推理的效果相差很大吗?

topDreamer avatar May 12 '23 04:05 topDreamer

您好,这个结果可以参照我们的期刊扩展https://github.com/Yunfan-Li/Twin-Contrastive-Learning ,效果相差不大,不过cluster head稍好一些

Yunfan-Li avatar May 12 '23 04:05 Yunfan-Li