我编写了一个函数来计算向量的熵,其中每个元素代表一个类的元素数量。
function x = Entropy(a)
t = sum(a);
t = repmat(t, [1, size(a, 2)]);
x = sum(-a./t .* log2(a./t));
end
e.g: a = [4 0]
, then entropy = -(0/4)*log2(0/4) - (4/4)*log2(4/4)
但对于上述函数,熵是NaN
当分裂是纯粹的,因为log2(0)
,如上例所示。纯分裂的熵应该为零。
数据量很大,如何解决对性能影响最小的问题?谢谢
我建议你创建自己的log2
功能
function res=mylog2(a)
res=log2(a);
res(isinf(res))=0;
end
这个功能,同时打破了log2
行为,可以在您的特定示例中使用,因为您将结果与日志内部相乘,从而使其为零。这不是“数学上正确的”,但我相信这就是您正在寻找的。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)