search

Yunfan-Li / Contrastive-Clustering

Code for the paper "Contrastive Clustering" (AAAI 2021)
MIT License
289 stars 92 forks source link

关于对比聚类的性能问题 #60

Open DavinciWu opened 2 weeks ago

DavinciWu commented 2 weeks ago

你好，非常感谢你对这项工作的开源，我在尝试这个项目的时候发现效果不尽人意，还请你多多指点： 我用自己的数据集（眼底彩照四个分类）进行实验，在3090上进行了1500轮的训练，训练损失从一开始的7.55降到5.62 然后 loss_instance从6.2降到4.2。 loss_cluster从1.98降到1.19 最终在验证集上评估的性能指标如下 NMI = 0.4193 ARI = 0.3484 F = 0.5242 ACC = 0.5338

Yunfan-Li commented 2 weeks ago

您好，可以考虑根据数据集的特性设计合适的数据增广策略