matlab 归一化互信息代码

MATLAB中的归一化互信息可以通过计算变量之间的互信息来实现。具体代码如下：

function [nmi] = normalized_mutual_information(x, y)
%计算归一化互信息
%输入：
%    x, y: 两个输入向量，长度相同
%输出：
%    nmi: 归一化互信息值

if nargin < 2
    error('请提供两个输入向量');
end

% 计算x和y的熵
Hx = entropy(x);
Hy = entropy(y);

% 计算x和y的联合熵
Hxy = entropy([x(:), y(:)]);

% 计算互信息
Ixy = Hx + Hy - Hxy;

% 计算归一化互信息
nmi = Ixy / sqrt(Hx * Hy);

function [H] = entropy(x)
%计算熵
%输入：
%    x: 输入向量
%输出：
%    H: 熵值

if isempty(x)
    H = 0;
else
    x = x(:);
    px = hist(x, unique(x)) / length(x);
    H = -sum(px .* log2(px));
end

使用时，只需要输入两个向量x和y即可计算它们的归一化互信息值。