matlab 评价聚类,MATLAB聚类有效性评价指标（外部）

MATLAB聚类有效性评价指标(外部)

前提：数据的真实标签已知！

1. 归一化互信息(Normalized Mutual information)

定义

程序

function MIhat = nmi(A, B)

%NMI Normalized mutual information

% A, B: 1*N;

if length(A) ~= length(B)

error('length( A ) must == length( B)');

end

N = length(A);

A_id = unique(A);

K_A = length(A_id);

B_id = unique(B);

K_B = length(B_id);

% Mutual information

A_occur = double (repmat( A, K_A, 1) == repmat( A_id', 1, N ));

B_occur = double (repmat( B, K_B, 1) == repmat( B_id', 1, N ));

AB_occur = A_occur * B_occur';

P_A= sum(A_occur') / N;<