update_hyperparams_doubletree: update hyperparameters

In zhenkewu/doubletree: Nested Latent Class Models for Domain-Adaptive and Semisupervised Learning of Cause-of-Deaths using Verbal Autopsy

update hyperparameters

Description

NB: argument redundancy may exist

Usage

update_hyperparams_doubletree(
  h_pau,
  levels,
  v_units,
  X,
  n,
  J,
  p1,
  p2,
  pL1,
  pL2,
  Fg1,
  Fg2,
  ind_obs_i,
  prob1,
  prob2,
  mu_gamma,
  mu_alpha,
  rmat,
  emat,
  dirich_mat,
  sigma_gamma,
  sigma_alpha,
  tau_1_t,
  tau_2_t,
  a1_t,
  b1_t,
  a2_t,
  b2_t,
  E_beta_sq,
  E_eta_sq,
  E_beta,
  E_eta,
  psi,
  g_psi,
  phi,
  g_phi,
  tau_1,
  tau_2,
  a1,
  b1,
  a2,
  b2,
  dmat,
  K,
  LD,
  tau_update_levels,
  update_hyper,
  quiet,
  do_tree1_update
)

Arguments

prob1, prob2, mu_gamma, mu_alpha, rmat, emat, dirich_mat, sigma_gamma, sigma_alpha, tau_1_t, tau_2_t, a1_t, b1_t, a2_t, b2_t	variational parameters
E_beta_sq, E_eta_sq, E_beta, E_eta	moments computed by update_vi_params_doubletree()
psi, g_psi, phi, g_phi, tau_1, tau_2	parameters updated by update_hyperparams_doubletree()
tau_update_levels	a numeric vector, specifies which levels of hyperparameters to update
update_hyper	Logical, TRUE or FALSE to indicate whether to update tau_1 and tau_2. This is computed at every iteration in fit_nlcm_doubletree()
quiet	default to FALSE, which prints intermediate updates of hyperparameters
do_tree1_update	TRUE to update mu_gamma,sigma_gamma, prob1,tau_1

Value

a list of updated hyperparameters: tau_1,tau_2,psi,g_psi,phi,g_phi, along with a new ELBO value.

See Also

Other Internal VI functions: update_vi_params_doubletree()

zhenkewu/doubletree documentation built on Oct. 21, 2023, 7:04 a.m.