EDA365电子论坛网

标题: #技术风云榜#分享列向量互信息计算通用MATLAB代码 [打印本页]

---

作者: UNerk    时间: 2020-11-11 14:52
标题: #技术风云榜#分享列向量互信息计算通用MATLAB代码

* v8 h7 f9 @) A, x' p6 _; P通用MATLAB代码
主函数main.m
' U3 i. X# Z2 R% ~5 ^0 v

• clc
• u1 = rand(4,1);
• u2 = [2;32;6666;5];
• wind_size = size(u1,1);
• mi = calmi(u1, u2, wind_size);
  % p$ }1 r) {! ]8 x# V0 L0 ^

0 y0 Z- w9 [; X! C& V; v! n: Fcalmi.m

9 ^8 A9 _- k* s5 S

• %计算两列向量之间的互信息
• %u1：输入计算的向量1
• %u2：输入计算的向量2
• %wind_size：向量的长度
• function mi = calmi(u1, u2, wind_size)
• x = [u1, u2];
• n = wind_size;
• [xrow, xcol] = size(x);
• bin = zeros(xrow,xcol);
• pmf = zeros(n, 2);
• for i = 1:2
•     minx = min(x(:,i));
•     maxx = max(x(:,i));
•     binwidth = (maxx - minx) / n;
•     edges = minx + binwidth*(0:n);
•     histcEdges = [-Inf edges(2:end-1) Inf];
•     [occur,bin(:,i)] = histc(x(:,i),histcEdges,1); %通过直方图方式计算单个向量的直方图分布
•     pmf(:,i) = occur(1:n)./xrow;
• end
• %计算u1和u2的联合概率密度
• jointOccur = accumarray(bin,1,[n,n]);  %（xi，yi）两个数据同时落入n*n等分方格中的数量即为联合概率密度
• jointPmf = jointOccur./xrow;
• Hx = -(pmf(:,1))'*log2(pmf(:,1)+eps);
• Hy = -(pmf(:,2))'*log2(pmf(:,2)+eps);
• Hxy = -(jointPmf(：）)'*log2(jointPmf(：）+eps);
• MI = Hx+Hy-Hxy;
• mi = MI/sqrt(Hx*Hy);
  

     
1 i! a7 s6 _; t, w

---

作者: ExxNEN    时间: 2020-11-11 15:21
列向量互信息计算通用MATLAB代码

---

欢迎光临 EDA365电子论坛网 (https://bbs.eda365.com/)

Powered by Discuz! X3.2