R/DDgetZ.a.R

In selenashuowang/jlsm: Joint Latent Space Model for Social Networks with Multivariate Attributes

DDgetZ.a.t.i.ia<-function(Y.i,Y.ia, Lambda.0,Lambda.1,Z.i,Z.a,p.lambda.1,alpha.1){
  

  B <- solve(diag(nrow(Lambda.1)) + 2 * Lambda.0 + 2 * Lambda.1)
  
  for (n in 1:nrow(Z.a)){
    
    
    dnj.ia <- t(Z.a[n,] - t(as.matrix(Z.i)))
    
    A.ia <-  1/sqrt(det(B)) * exp(- alpha.1) * exp(apply(dnj.ia, 1, function(x) x %*% B %*% x))
    
    f1znTo.ia <- - 2 * B %*% colSums(dnj.ia / (1+A.ia))
    f2znTo.ia <- - 2 * sum(1 / (1 + A.ia)) * B + 
      4 * B %*% (t(dnj.ia / (2 + 1 / A.ia + A.ia)) %*% dnj.ia) %*% B
    
    #
    numZnT <-  colSums(as.matrix(Z.i)* (Y.ia[,n]))+ .5*Z.a[n, ] %*% (f2znTo.ia) - .5*t(f1znTo.ia)
    denZnT <- ( 1 / (2* p.lambda.1^2) + sum((Y.ia[,n]))) * diag(nrow(Lambda.1))  +.5*f2znTo.ia
    
    Z.a[n, ]<- numZnT %*% solve(denZnT)
  }
  
  Z.a
}