第一篇论文实验
dgn伪代码
预处理部分:(即这部分我不用改)
1.pca降维
2.拉普拉斯特征
对比学习部分:
正常对比学习(即这部分我不用改)
聚类部分:
聚类
用两层的cluster_projector
nn.Linear(self.d,self.d)
nn.ReLU
nn.Linear(self.d,self.cluster_num)
nn.Softmax(dim=1)
得到聚类/聚类中心(这部分改)
(如何设计的)
得到聚类loss
DQN部分:
暂时我需要删掉DQN部分,只计算两个损失
第一次优化:实例级别对比,NMI=48,在迭代30次左右后.
其中,取了instance contrastive +wss作为对比损失.
初始化可以用xavieruniform初始化