MATLAB实例：为匹配真实标签，对训练得到的标签进行调整

作者: 你最珍贵10500288
来源: 51数据库
2020-08-06

MATLAB实例：为匹配真实标签，对训练得到的标签进行调整

作者：凯鲁嘎吉 - 博客园 http://www.cnblogs.com/kailugaji/

1. MATLAB程序

munkres.m

见：MATLAB实例：Munkres指派算法

label_map.m

function [ new_label ] = label_map( label, gnd )
%为匹配真实标签，对标签重新调整
K = length(unique(gnd));
cost_mat = zeros(K,K);
for i=1:K
    idx = find(label==i);
    for j=1:K        
        cost_mat(i,j) = length(find(gnd(idx)~=j));
    end
end
[assignment,cost] = munkres(cost_mat);
[assignedrows,dum]=find(assignment');
new_label = label;
for i=1:K
    idx = find(label==i);
    new_label(idx) = assignedrows(i);
end

2. 结果

>> label=[1 1 2 1 1 2 2 2 3 2 2 3 1 3 3 2 3];
>> gnd=[2 2 2 2 2 2 3 3 3 3 3 3 1 1 1 1 1 ];
>> [ new_label ] = label_map( label, gnd )

new_label =

     2     2     3     2     2     3     3     3     1     3     3     1     2     1     1     3     1