scikit-learn中的nmi有时无法正常工作

时间:2019-07-12 17:25:28

标签: scikit-learn cluster-analysis nmi

我有两个检测到的社区,我想计算它们的nmi:

   realNMI
   Out[5]: 
   array([0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 1., 1., 
   0.,0., 1., 0., 1., 0., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.])

和:

   resultNMI
   Out[12]: 
   array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 
   0.,0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.])

我想计算nmi:

   from sklearn.metrics.cluster import normalized_mutual_info_score
   normalized_mutual_info_score(realNMI, resultNMI)
   Out[13]: 1.0

为什么会这样?

我尝试过:

   normalized_mutual_info_score(realNMI.astype(int),re.astype(int))

但仍然得到1.0

0 个答案:

没有答案