compute_auxiliary_trait_elbo: Auxiliary Trait ELBO

View source: R/cavi_auxiliary_traits.R

compute_auxiliary_trait_elboR Documentation

Auxiliary Trait ELBO

Description

Compute the contribution of auxiliary traits to the Evidence Lower Bound (ELBO) of a PLVM given the approximate posterior distribution for loadings and latent traits

Usage

compute_auxiliary_trait_elbo(
  manifest_trait_df,
  metadata,
  auxiliary_traits,
  loading_expectation,
  latent_trait_expectation,
  precision,
  loading_outer_expectation,
  latent_trait_outer_expectation,
  n_samples = 1000,
  random_seed = NULL,
  perform_checks = TRUE
)

Arguments

manifest_trait_df

The data frame of manifest traits. If included allows for a non-exhaustive series of checks comparing metadata provided to that implicit in the manifest traits.

metadata

A data frame. Contains all the metadata required to map a set of manifest traits to the PLVM auxiliary traits..

auxiliary_traits

An NxD' matrix of real numbers. The auxiliary traits.

loading_expectation

A D'xL matrix of real numbers, The expected loading matrix.

latent_trait_expectation

An NxL matrix of real values. The expected individual specific latent traits.

precision

A P-dimensional vector of positive real values. The precision with which auxiliary traits are observed.

loading_outer_expectation

A LxLxD' array. The expected outer products of the expected loading matrix.

latent_trait_outer_expectation

A LxLxN array. The expected outer product of individual specific latent traits.

n_samples

A positive integer. The number of independent samples drawn to obtain the Monte Carlo estimate.

random_seed

A single value, interpreted as an integer, or NULL.

perform_checks

Logical. Check if function inputs are specified correctly.

Value

A scalar. The contribution of Auxiliary traits to the ELBO.


jpmeagher/vbar documentation built on Nov. 22, 2022, 5:48 a.m.