IS散度的NMF算法matlab代码

以下是IS散度的NMF算法的MATLAB代码：

function [W,H] = nmf_isdiv(X,k,max_iter)
% NMF using IS divergence
% X: input matrix
% k: number of basis vectors
% max_iter: maximum number of iterations
% W: basis matrix
% H: coefficient matrix

% initialize W and H with random values
[n,m] = size(X);
W = rand(n,k);
H = rand(k,m);

% set small value for avoiding division by zero
eps = 1e-16;

% iterate until convergence or max_iter
for iter = 1:max_iter
    % update H
    for j = 1:m
        H(:,j) = H(:,j) .* (W' * (X(:,j) ./ (W * H(:,j) + eps))) ./ sum(W,1)';
    end
    
    % update W
    for i = 1:n
        W(i,:) = W(i,:) .* ((X(i,:) ./ (W(i,:) * H + eps)) * H') ./ sum(H,2)';
    end
    
    % calculate IS divergence
    div = sum(sum(X .* log(X./(W*H+eps)) - X + W*H));
    
    % check for convergence
    if iter > 1 && abs(div-prev_div) < 1e-6
        break
    end
    prev_div = div;
end

end

使用方法：

假设X为输入矩阵，k为基向量的数量，max_iter为最大迭代次数。调用函数nmf_isdiv(X,k,max_iter)即可进行IS散度的NMF算法计算，返回值为W和H矩阵。