getSampledBatchShift {batchmix}

Get sampled batch shift

Description

Given an array of sampled batch mean shifts from the “mixtureModel“ function, acquire a tidy version ready for “ggplot2“ use.

Usage

getSampledBatchShift(
  sampled_batch_shift,
  B = dim(sampled_batch_shift)[2],
  P = dim(sampled_batch_shift)[1],
  R = dim(sampled_batch_shift)[3],
  thin = 1
)

Arguments

sampled_batch_shift

A 3D array of sampled batch mean shifts.

B

The number of batches present. Defaults to the number of columns in the batch mean matrix from the first sample.

P

The dimension of the batch mean shifts. Defaults to the number of rows in the batch mean matrix from the first sample.

R

The number of iterations run. Defaults to the number of slices in the sampled batch mean array.

thin

The thinning factor of the sampler. Defaults to 1.

Value

A data.frame of three columns; the parameter, the sampled value and the iteration.

Examples

# Data in matrix format
X <- matrix(c(rnorm(100, 0, 1), rnorm(100, 3, 1)), ncol = 2, byrow = TRUE)

# Observed batches represented by integers
batch_vec <- sample(seq(1, 5), size = 100, replace = TRUE)

# MCMC iterations (this is too low for real use)
R <- 100
thin <- 5

# MCMC samples
samples <- runBatchMix(X, R, thin, batch_vec, "MVN")

batch_shift_df <- getSampledBatchShift(samples$batch_shift,
  R = R,
  thin = thin
)