社区发现评估指标-NMI

最新推荐文章于 2024-04-08 10:29:24 发布

开开_王子

最新推荐文章于 2024-04-08 10:29:24 发布

阅读量1.2w

点赞数 1

分类专栏：算法与数据结构文章标签：复杂网络数据挖掘

本文链接：https://blog.csdn.net/kan2281123066/article/details/73825009

版权

NMI（标准化互信息）是用于评估聚类效果，特别是社区发现任务中的重要指标。它衡量了两个聚类结果的相似度，值域在0到1之间，值越大表示聚类准确性越高。在Python中实现NMI得到的结果为0.36456。

摘要由CSDN通过智能技术生成

1、介绍
NMI(Normalized Mutual Information)标准化互信息，常用在聚类中，度量两个聚类结果的相近程度。是社区发现(community detection)的重要衡量指标，基本可以比较客观地评价出一个社区划分与标准划分之间相比的准确度。NMI的值域是0到1，越高代表划分得越准。

2、python代码

# coding=utf-8
import numpy as np
import math
def NMI(A,B):
    # len(A) should be equal to len(B)
    total = len(A)
    A_ids = set(A)
    B_ids = set(B)
    #Mutual information
    MI = 0
    eps = 1.4e-45
    for idA in A_ids: