cdlib.evaluation.normalized_mutual_information

cdlib.evaluation.normalized_mutual_information(first_partition: object, second_partition: object) → MatchingResult

两个聚类之间的归一化互信息。

归一化互信息（NMI）是互信息（MI）得分的归一化，用于将结果缩放到0（无互信息）和1（完全相关）之间。在此函数中，互信息通过sqrt(H(labels_true) * H(labels_pred))进行归一化。

Parameters:

• first_partition – NodeClustering 对象

• second_partition – NodeClustering 对象

Returns:

匹配结果对象

Example:

>>> from cdlib import evaluation, algorithms
>>> import networkx as nx
>>> g = nx.karate_club_graph()
>>> louvain_communities = algorithms.louvain(g)
>>> leiden_communities = algorithms.leiden(g)
>>> evaluation.normalized_mutual_information(louvain_communities,leiden_communities)