facebookresearch / deepcluster

Deep Clustering for Unsupervised Learning of Visual Features
Other
1.68k stars 324 forks source link

about NMI #89

Open chester-w-xie opened 2 years ago

chester-w-xie commented 2 years ago

In your code, the value of NMI is calculated using the cluster labels generated by the two iterations, unlike in most articles, which use the predicted labels with the true labels to calculate NMI and several other metrics. I tried to modify your code to adopt the latter approach for the NMI calculation and found that the results became worse.