ARD_mcmc {BMIselect} | R Documentation |
ARD MCMC Sampler for Multiply-Imputed Regression
Description
Implements Bayesian variable selection using the Automatic Relevance Determination (ARD) prior across multiply-imputed datasets. The ARD prior imposes feature-specific shrinkage by placing a prior proportional to inverse of precision of each coefficient.
Usage
ARD_mcmc(
X,
Y,
intercept = TRUE,
nburn = 4000,
npost = 4000,
seed = NULL,
verbose = TRUE,
printevery = 1000,
chain_index = 1
)
Arguments
X |
A 3-D array of predictors with dimensions |
Y |
A matrix of outcomes with dimensions |
intercept |
Logical; include an intercept? Default |
nburn |
Integer; number of burn-in MCMC iterations. Default |
npost |
Integer; number of post-burn-in samples to retain. Default |
seed |
Integer or |
verbose |
Logical; print progress messages? Default |
printevery |
Integer; print progress every this many iterations. Default |
chain_index |
Integer; index of this MCMC chain (for labeling messages). Default |
Value
A named list
with components:
post_beta
Array
npost × D × p
of sampled regression coefficients.post_alpha
Matrix
npost × D
of sampled intercepts (if used).post_sigma2
Numeric vector length
npost
, sampled residual variances.post_psi2
Matrix
npost × p
of sampled precision parameters for each coefficient.post_fitted_Y
Array
npost × D × n
of posterior predictive draws (with noise).post_pool_beta
Matrix
(npost * D) × p
of pooled coefficient draws.post_pool_fitted_Y
Matrix
(npost * D) × n
of pooled predictive draws (with noise).hat_matrix_proj
Matrix
D × n × n
of averaged projection hat-matrices. To avoid recalculate for estimating degree of freedom.
Examples
sim <- sim_B(n = 100, p = 20, type = "MAR", SNP = 1.5, corr = 0.5,
low_missing = TRUE, n_imp = 5, seed = 123)
X <- sim$data_MI$X
Y <- sim$data_MI$Y
fit <- ARD_mcmc(X, Y, nburn = 100, npost = 100)