Question

通过以下程序，我会收到错误：

solve.default(Sigma0[cs.idx, cs.idx]) : 'a' is 0-diml

但是，当我逐步检查em()函数时，我的意思是，没有函数的逐句，solve()内没有错误。所以我很困惑，迫切需要帮助，谢谢！

###----------------------------------------------------------------
### Maximal Likelihood estimation of mean and covariance 
###  for multivariate normal distribution by EM algorithm,
###             for demonstration purposes only
###----------------------------------------------------------------

em<-function(xdata,mu0,Sigma0){
  n<-nrow(xdata)
  p<-ncol(xdata)

   err<-function(mu0,Sigma0,mu1,Sigma1){
    th0<-c(mu0,as.vector(Sigma0))
    th1<-c(mu1,as.vector(Sigma1))
    sqrt(sum((th0-th1)*(th0-th1)))
   }

  mu1<-mu0+1
  Sigma1<-Sigma0+1
  while(err(mu0,Sigma0,mu1,Sigma1)>1e-6){
     mu1<-mu0
     Sigma1<-Sigma0

     zdata<-xdata
     Ai<-matrix(0,p,p)
    for(i in 1:n){
     if(any(is.na(xdata[i,]))){
      zi<-xdata[i,]
      na.idx<-(1:p)[is.na(zi)]
      cs.idx<-(1:p)[-na.idx]
      Sigma012<-Sigma0[na.idx,cs.idx,drop=FALSE]
      Sigma022.iv<-solve(Sigma0[cs.idx,cs.idx])
      zdata[i,na.idx]<-mu0[na.idx]+(Sigma012%*%Sigma022.iv)%*%(zi[cs.idx]-mu0[cs.idx])
      Ai[na.idx,na.idx]<-Ai[na.idx,na.idx]+Sigma0[na.idx,na.idx]-Sigma012%*%Sigma022.iv%*%t(Sigma012)
       }
     }
    mu0<-colMeans(zdata)
    Sigma0<-(n-1)*cov(zdata)/n+Ai/n
   }
 return(list(mu=mu0,Sigma=Sigma0))
}

##A simulation example
library(MASS)
set.seed(1200)
p=3
mu<-c(1,0,-1)
n<-1000
Sig <- matrix(c(1, .7, .6, .7, 1, .4, .6, .4, 1), nrow = 3)
triv<-mvrnorm(n,mu,Sig)
misp<-0.2   #MCAR probability
misidx<-matrix(rbinom(3*n,1,misp)==1,nrow=n)
triv[misidx]<-NA 

#exclude the cases whose entire elements were missed
er<-which(apply(apply(triv,1,is.na),2,sum)==p)   
if(length(er)>=1) triv<-triv[-er,]

#initial values
mu0<-rep(0,p)
Sigma0<-diag(p)
system.time(rlt<-em(triv,mu0,Sigma0))

#a better initial values
mu0<-apply(triv,2,mean,na.rm=TRUE)
nas<-is.na(triv)
na.num<-apply(nas,2,sum)
zdata<-triv
zdata[nas]<-rep(mu0,na.num)
Sigma0<-cov(zdata)

system.time(rlt<-em(triv,mu0,Sigma0))

Answer 1

您的er<-which(apply(apply(triv,1,is.na),2,sum)==)代码无效。如上面的评论所述，您希望删除完整的NA个案例。如果是这样，er<-which(apply(apply(triv,1,is.na),2,sum)==ncol(triv))是正确的代码段。

当传递给NA时，triv中仍存在完整的em案例，就会发生错误。在某些时候，cs.idx为空，因此Sigma0[cs.idx,cs.idx]也为空，这将由错误消息反映出来。

但是，如果应用上述更正，一切运行正常：

> system.time(rlt<-em(triv,mu0,Sigma0))
   user  system elapsed 
   0.46    0.00    0.47 
> rlt
$mu
[1]  0.963058487 -0.006246175 -1.024260183

$Sigma
          [,1]      [,2]      [,3]
[1,] 0.9721301 0.6603700 0.5549126
[2,] 0.6603700 1.0292379 0.3745184
[3,] 0.5549126 0.3745184 0.9373208

使用R中的solve（）在函数内运行错误？

1 个答案: