1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
set.seed(123)
data(longley)
data <- list(gnp=longley$GNP, employed=longley$Employed, n=nrow(longley))
modfile <- tempfile()
writeLines("
model{
for (i in 1:n){
employed[i] ~ dnorm(mu[i], tau)
mu[i] <- alpha + beta*gnp[i]
}
alpha ~ dnorm(0, 0.00001)
beta ~ dnorm(0, 0.00001)
sigma ~ dunif(0,1000)
tau <- pow(sigma,-2)
}", con=modfile)
inits <- function(){
list(alpha=rnorm(1,0,1),beta=rnorm(1,0,1),sigma=runif(1,0,3))
}
params <- c('alpha','beta','sigma', 'mu')
out <- jags.basic(data = data, inits = inits, parameters.to.save = params,
model.file = modfile, n.chains = 3, n.adapt = 100, n.iter = 100,
n.burnin = 50, n.thin = 2, verbose=FALSE)
ref <- readRDS("jagsbasic_reference_fit.Rds")
expect_equal(out, ref)
# Saved model and reordered parameter names------------------------------------
set.seed(123)
params <- c('beta', 'alpha', 'sigma', 'mu')
out <- jags.basic(data = data, inits = inits, parameters.to.save = params,
model.file = modfile, n.chains = 3, n.adapt = 100, n.iter = 100,
n.burnin = 50, n.thin = 2, verbose=FALSE, save.model=TRUE)
ref <- readRDS("jagsbasic_ref_saved.Rds")
expect_equal(names(out), names(ref))
out$model <- ref$model
expect_equal(out, ref)
# Update-----------------------------------------------------------------------
out2 <- update(out, n.iter=100, n.thin = 2, verbose=FALSE)
expect_equal(nrow(out2$samples[[1]]), 50)
ref <- readRDS('jagsbasic_ref_update.Rds')
expect_equal(names(out2), names(ref))
out2$model <- ref$model
expect_equal(out2, ref)
# Error if seed is set---------------------------------------------------------
expect_error(jags.basic(data = data, inits = inits, parameters.to.save = params,
model.file = modfile, n.chains = 3, n.adapt = 100, n.iter = 100,
n.burnin = 50, n.thin = 2, verbose=FALSE, save.model=TRUE, seed=123))
# Parallel---------------------------------------------------------------------
at_home <- identical( Sys.getenv("AT_HOME"), "TRUE" )
if(parallel::detectCores() > 1 & at_home){
set.seed(123)
params <- c('beta', 'alpha', 'sigma', 'mu')
out <- jags.basic(data = data, inits = inits, parameters.to.save = params,
model.file = modfile, n.chains = 3, n.adapt = 100, n.iter = 100,
n.burnin = 50, n.thin = 2, verbose=FALSE, save.model=TRUE, parallel=TRUE)
ref <- readRDS("jagsbasic_ref_saved.Rds")
out$n.cores <- NULL
expect_equal(names(out), names(ref))
out$model <- ref$model
expect_equal(out, ref)
out2 <- update(out, n.iter=100, n.thin = 2, verbose=FALSE)
ref <- readRDS('jagsbasic_ref_update.Rds')
expect_equal(names(out2), names(ref))
out2$model <- ref$model
expect_equal(out2, ref)
}
# Verbose---------------------------------------------------------------------
co <- capture.output(jags.basic(data = data, inits = inits, parameters.to.save = params,
model.file = modfile, n.chains = 3, n.adapt = 100, n.iter = 100,
n.burnin = 50, n.thin = 2, verbose=TRUE, save.model=TRUE)
)
test <- any(sapply(co, function(x) grepl("MCMC took", x)))
expect_true(test)
|