generateInitialLabels: Generate initial labels

In batchmix: Semi-Supervised Bayesian Mixture Models Incorporating Batch Correction

Generate initial labels

Description

For simulated data, generates an initial labelling for sampling.

Usage

generateInitialLabels(alpha, K, fixed, labels = NULL)

Arguments

alpha	The mass in the stick breaking prior
K	The number of classes available.
fixed	The vector of 0s and 1s indicating which labels are to be held fixed.
labels	The initial labelling. Defaults to NULL.

Value

An N vector of labels.

Examples

N <- 500
P <- 2
K <- 2
B <- 5
mean_dist <- 4
batch_dist <- 0.3
cluster_means <- seq(1, K) * mean_dist
batch_shift <- rnorm(B, mean = batch_dist, sd = batch_dist)
std_dev <- rep(2, K)
batch_var <- rep(1.2, B)
cluster_weights <- rep(1 / K, K)
batch_weights <- rep(1 / B, B)

my_data <- generateBatchData(
  N,
  P,
  cluster_means,
  std_dev,
  batch_shift,
  batch_var,
  cluster_weights,
  batch_weights
)

initial_labels <- generateInitialLabels(1, K, my_data$fixed)

batchmix documentation built on May 29, 2024, 2:14 a.m.