r语言 - JAGS 后计算和 ifelse/step



我对JAGS比较陌生,并且正在通过R包jagsUI运行它。我正在构建占用模型,但想在进行时总结结果。所以我有一个 0 和 1 的矩阵:

mat1 <- matrix(rbinom(10*10,1,.5),10,10)
y=mat1

我想通过以下模型运行:

# Bundle data and summarize data bundle
str( win.data <- list(y = mat1, M = nrow(mat1), T = ncol(mat1)) )
# Specify model in BUGS language
sink("model.txt")
cat("
model {
# Priors
   psi0 ~ dunif(0, 1)
   p ~ dunif(0, 1)
for(t in 1:(T-1)){
  rho[t] ~ dunif(-1,1)
}
beta0 ~ dnorm(0, 0.1)
# Likelihood
   for (i in 1:M) {    # Loop over sites
      z[i,1] ~ dbern(psi0)         # State model
      y[i,1] ~ dbern(z[i,1]*p)
      for (j in 2:T) {    # Loop over replicate surveys
          logit(psi[i,j])<- beta0 + rho[j-1]*z[i,j-1]
          z[i,j] ~ dbern(psi[i,j])
          y[i,j] ~ dbern(z[i,j]*p)  # Observation model
       }
   }
# Derived quantities
    coln[i,j] <- ifelse(z[i,j]-z[i,j-1]==1,1,0) # colonized
    ext[i,j] <- ifelse(z[i,j-1]-z[i,j]==1,1,0) # went extinct
    tot.coln[,j] <- sum(coln[,j]) # sum of colonized each survey
    tot.ext[,j] <- sum(ext[,j]) # sum of extinctions each survey
    Nocc[,j] <-  sum(z[,j]) # total sites occupied each survey
    coln.rate[,j] <- tot.coln[,j]/Nocc[,j]
    ext.rate[,j] <- tot.ext[,j]/Nocc[,j]
}
",fill = TRUE)
sink()
# Initial values
zst <- apply(y, 1, max, na.rm=TRUE)       # Avoid data/model/inits conflict
y<- as.matrix(y)
zst<- y
inits <- function(){list(z = zst)}
# Parameters monitored
params <- c("psi0", "p", "beta0", "coln.rate", "ext.rate")
# MCMC settings
ni <- 2000   ;   nt <- 1   ;   nb <- 1000   ;   nc <- 3
# Call JAGS and summarize posteriors
library(jagsUI)
fm <- jags(win.data, inits, params, "model.txt", n.chains = nc,
           n.thin = nt, n.iter = ni, n.burnin = nb)
print(fm, dig = 3)

模型运行,除了"# 派生数量"之后的部分。基本上,我想计算每次调查中从 0 到 1 和从 1 到 0 的变化率。我对为什么它不起作用的一些想法。1( z[i,j] 不是真正的 0 和 1。 2( 计算不应在派生量下进行。3(JAGS手册中的ifelse没有按照我的想法做。

我还尝试使用"step"函数将派生数量之后的前两行替换为:

coln[i,j] <- step(z[i,j]-z[i,j-1]-0.5) # colonized
ext[i,j] <- step(z[i,j-1]-z[i,j]-0.5) # went extinct

但是那里没有运气。有什么想法吗?

您正在索引i并在此处j,而无需循环访问它们。要完成这项工作,您需要在另一个嵌套的 for 循环中设置它。此外,您的灭绝计算不正确。

for(j in 2:T){
    for(i in 1:M){
    coln[i,j-1] <- ifelse(z[i,j]-z[i,j-1]==1,1,0) # colonized
    ext[i,j-1] <- ifelse(z[i,j]-z[i,j-1]==-1,1,0) # went extinct
}
    tot.coln[j-1] <- sum(coln[,j-1]) # sum of colonized each survey
    tot.ext[j-1] <- sum(ext[,j-1]) # sum of extinctions each survey
    Nocc[j-1] <-  sum(z[,j-1]) # total sites occupied each survey
    coln.rate[j-1] <- tot.coln[j-1]/Nocc[j-1]
    ext.rate[j-1] <- tot.ext[j-1]/Nocc[j-1]
}

最新更新