Question

用于虚构数据的算法的代码：我正在努力将其转换为MATLAB，但却在循环中运行的计算中苦苦挣扎。任何帮助将不胜感激。

data <- c(-0.39, 0.12, 0.94, 1.67, 1.76, 2.44, 3.72, 
       4.28, 4.92, 5.53, 0.06, 0.48, 1.01, 1.68, 1.80, 
       3.25, 4.12, 4.60, 5.28, 6.22) 
pi <- 0.546  
sigmas1 <- 0.87 
sigmas2 <- 0.77 
mu1 <- numeric(0) 
mu2 <- numeric(0) 
r   <- numeric(0) 
R1  <- matrix (0 ,20 ,100) 
mu1[1] <- 4.62 
mu2[1] <- 1.06

for(j in 1:100){
  for ( i in 1:20){
    r [i] <- pi * dnorm (data[i] , mu2[j], sigmas2^(1/2))/((1-    pi)*dnorm(data[i], 
        mu1[j], sigmas1^(1/2))+ pi*dnorm(data[i], mu2[j], sigmas2^(1/2))) 
R1[i, j] <- r[i] 
}
r
mu1[j+1] <- sum((1-r)*data)/sum(1-r)
mu2[j+1] <- sum(r*data)/sum(r) 
Muu1 <- mu1[j+1]
Muu2 <- mu2[j+1]
}

Muu1
Muu2
x11() 
layout(matrix(c(1, 2))) 
plot(mu1, type="l", main="", xlab="EM Iteration for the Fictitious Data")
plot(mu2, type="l", main="", xlab='EM Iteration for the Fictitious Data')

Answer 1

R的dnorm函数的MATLAB等价物是normpdf。参数与R中的相同：

normpdf(X,mu,sigma)

这样就可以很容易地调整for循环。由于normpdf函数允许向量作为输入，因此您可以转储内部for循环并使用向量化方法。请记住，*和/是MATLAB中的矩阵乘法和除法。要获取元素操作符，请改用.*和./。

请注意，在MATLAB中，最好预先分配所有变量。由于mu1和mu2从1到100，但在每个步骤中设置值mu[j+1]，其大小为1x101。对于r和R1，我认为大小很明显。

总之，这将提供以下代码：

data = [-0.39, 0.12, 0.94, 1.67, 1.76, 2.44, 3.72,...
    4.28, 4.92, 5.53, 0.06, 0.48, 1.01, 1.68, 1.80,...
    3.25, 4.12, 4.60, 5.28, 6.22];

pi=0.546;
sigmas1 = 0.87;
sigmas2 = 0.77; 
mu1 = zeros(1,101);
mu2 = zeros(1,101);
r = zeros(1,20);
R1 = zeros(20,100);
mu1(1) = 4.62;
mu2(1) = 1.06;

for j=1:100
    r= pi*normpdf(data,mu2(j),sigmas2^(1/2)) ./ ...
        ((1-pi)*normpdf(data,mu1(j),sigmas1^(1/2)) + ...
         pi*normpdf(data,mu2(j),sigmas2^(1/2)));
    R1(:,j) = r;
    mu1(j+1) = sum((1-r).*data)/sum(1-r);
    mu2(j+1) = sum(r.*data)/sum(r);
end

figure;
subplot(1,2,1);
plot(mu1);
subplot(1,2,2);
plot(mu2);

如果这对您不起作用，或者您对代码有任何疑问，请随时发表评论。

将R代码转换为MATLAB

1 个答案: