elbo_cluster {survregVB} | R Documentation |
Calculates the variational Bayes convergence criteria, evidence lower
bound (ELBO), optimized in survregVB.frailty.fit
.
Description
Calculates the variational Bayes convergence criteria, evidence lower
bound (ELBO), optimized in survregVB.frailty.fit
.
Usage
elbo_cluster(
y,
X,
delta,
alpha_0,
omega_0,
mu_0,
v_0,
lambda_0,
eta_0,
alpha,
omega,
mu,
Sigma,
tau,
sigma,
lambda,
eta,
expectation_b,
cluster
)
Arguments
y |
A vector of observed log-transformed survival times. |
X |
A design matrix including covariates with first column of ones to represent the intercept. |
delta |
A binary vector indicating right censoring. |
alpha_0 |
The shape hyperparameter |
omega_0 |
The shape hyperparameter |
mu_0 |
Hyperparameter |
v_0 |
The precision (inverse variance) hyperparameter |
lambda_0 |
The shape hyperparameter |
eta_0 |
The scale hyperparameter |
alpha |
The shape parameter |
omega |
The scale parameter |
mu |
Parameter |
Sigma |
Parameter |
tau |
Parameter |
sigma |
Parameter |
lambda |
The shape parameter |
eta |
The scale parameter |
expectation_b |
The expected value of b. |
cluster |
A numeric vector indicating the cluster assignment for each observation. |
Value
The evidence lower bound (ELBO).