使用与地面实况定义相同的颜色为集群着色以进行可视化

Posted 2023-03-12

技术标签:

【中文标题】使用与地面实况定义相同的颜色为集群着色以进行可视化【英文标题】：Coloring the cluster with same colors as defined for ground truth for visualization 【发布时间】：2020-03-19 17:30:16 【问题描述】：

示例：（考虑平台 = MATLAB）

Ground_Truth_Indices = [ 1, 1, 1, 2, 2, 2, 3, 3, 3];

对于 GT 中的每个唯一索引，我定义了一个颜色数组。

Color_Array = [ 0, 255, 0;  255, 0, 0;  0, 0, 255]; %assuming (in this eg.) the max. cluster size is 3

接下来，我使用了一个聚类算法（在我的例子中是 DBSCAN），它给出了以下索引：

Clustered_Indices = [2, 2, 2, 3, 3, 3, 1, 1, 1];

现在，我需要将结果与基本事实一起可视化。但聚类后得到的指标与ground truth指标不同。

因此，根据定义的颜色数组，在可视化过程中，我不会得到与地面实况相同的颜色模式并获得聚类。有什么解决方案可以使两种颜色保持一致吗？

Figure with ground truth and obtained clusters

上面的图链接也说明了这一点（不是 MatLab 绘图！为了说明目的而创建），其中集群 1 在基本实况以及获得的集群结果中应该具有相同的颜色。但是，这里不是这样，因为定义了与颜色数组相关的索引号。

注意：聚类后得到的索引不能预先定义，取决于聚类算法和聚类输入。

【问题讨论】：

能否请您edit 您的帖子并添加一些数字来显示您期望获得什么集群以及您正在获得什么？附带说明一下，MATLAB 中的 cmets 以 % 而不是 # 开头。 【参考方案1】：

灵感来自这篇文章的答案： How can I match up cluster labels to my 'ground truth' labels in Matlab，我的问题有以下解决方案代码：

N = length(Ground_Truth_Indices);

cluster_names = unique(Clustered_Indices);
accuracy = 0;
maxInd = 1;

perm = perms(unique(Ground_Truth_Indices));
[perm_nrows perm_ncols] = size(perm);

true_labels = Ground_Truth_Indices;

for i=1:perm_nrows
    flipped_labels = zeros(1,N);
    for cl = 1 : perm_ncol
        flipped_labels(Clustered_Indices==cluster_names(cl)) = perm(i,cl);
    end

    testAcc = sum(flipped_labels == Ground_Truth_Indices')/N;
    if testAcc > accuracy
        accuracy = testAcc;
        maxInd = i;
        true_labels = flipped_labels;
    end

end

其中“true_labels”包含根据变量“Ground_Truth_Indices”重新排列的变量“Clustered_Indices”的标签。