用户登录
用户注册

分享至

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

3. 参考文献

[1] Hua J, Li C. Distributed variational Bayesian algorithms over sensor networks[J]. IEEE Transactions on Signal Processing, 2015, 64(3): 783-798.

[2] Junhao Hua. Distributed Variational Bayesian Algorithms. Github, 2017.

[3] MATLAB实例:Munkres指派算法

原文链接:http://www.cnblogs.com/kailugaji/p/11771226.html

软件
前端设计
程序设计
Java相关