如何在 R 中使用不同的 set.seed() 多次运行相同的模型？

Question

我想用不同的种子运行以下模型 3 次。例如，下面的模型使用种子314159运行

set.seed(314159)
    x <- c(11, 5, 2, -5, 7, 2, -11, 9, -5, -5, -4, 17, 2, -10, -11, -10,
           -4, 2, 1, 13)
    a <- 0.1
    b <- 0.1
    c <- 0
    d <- 100^2

    M <- 1e3
    sample <- array(NA, dim=c(M,2))
    mu <- mean(x)
    sig2 <- var(x)
    for( m in 1:M ){
      mu <- rnorm(1, (length(x) + 1/d)^(-1) * (sum(x) + c/d),
                  sqrt( sig2/(length(x) + 1/d) ))
      sig2 <- rigamma(1, .5*length(x)+a+.5,
                      .5*sum( (x-mu)^2 ) + 1/(2*d)*(mu-c)^2 + b )
      sample[m,] <- c(mu,sig2)
    }
    plot( density( sample[,1] ))
    plot( density( sample[,2] ))

如果我想为种子 523626 和 626789 运行相同的模型，我可以使用任何 for 循环吗？ 任何帮助表示赞赏？

Answer 1

将代码放入函数中

apply_fun <- function() {
  x <- c(11, 5, 2, -5, 7, 2, -11, 9, -5, -5, -4, 17, 2, -10, -11, -10,-4, 2, 1, 13)
  a <- 0.1
  b <- 0.1
  c <- 0
  d <- 100^2
  M <- 1e3
  sample <- array(NA, dim=c(M,2))
  mu <- mean(x)
  sig2 <- var(x)
  for( m in 1:M ){
    mu <- rnorm(1, (length(x) + 1/d)^(-1) * (sum(x) + c/d),
            sqrt( sig2/(length(x) + 1/d) ))
    sig2 <- rigamma(1, .5*length(x)+a+.5,
                .5*sum( (x-mu)^2 ) + 1/(2*d)*(mu-c)^2 + b )
    sample[m,] <- c(mu,sig2)
  }
   plot( density( sample[,1] ))
   plot( density( sample[,2] ))
}

然后在每个种子值上使用lapply

output <- lapply(c(314159, 523626, 626789), function(x) {set.seed(x);apply_fun()})

rigamma 在哪里

rigamma = function(n, a, b) return(1/rgamma(n, shape = a, rate = b))

Answer 2

你可以循环遍历种子

library(LearnBayes)
seeds <- c(314159,523626,626789)

for (seed in seeds) {
  set.seed(seed)
  x <- c(11, 5, 2, -5, 7, 2, -11, 9, -5, -5, -4, 17, 2, -10, -11, -10,
    -4, 2, 1, 13)
  a <- 0.1
  b <- 0.1
  c <- 0
  d <- 100^2

  M <- 1e3
  sample <- array(NA, dim=c(M,2))
  mu <- mean(x)
  sig2 <- var(x)
  for( m in 1:M ){
    mu <- rnorm(1, (length(x) + 1/d)^(-1) * (sum(x) + c/d),
      sqrt( sig2/(length(x) + 1/d) ))
    sig2 <- rigamma(1, .5*length(x)+a+.5,
      .5*sum( (x-mu)^2 ) + 1/(2*d)*(mu-c)^2 + b )
    sample[m,] <- c(mu,sig2)
  }
  plot( density( sample[,1] ))
  plot( density( sample[,2] ))
}

^{由reprex package (v0.3.0) 于 2020-01-12 创建}