train_NBFA2: Train a negative binomial factor analysis model

In gregorp90/RStan-package-test: Copula-Based Factor Analysis

Description

This function trains a negative binomial factor analysis model in Stan. Each variable has its own scaling parameter.

Usage

train_NBFA2(train, gp_train, nfac = 2, ...)

Arguments

train	data frame. The training data set, consisting of counts. Columns represent variables, rows represent observations.
gp_train	vector. Groups for each observation.
nfac	numeric. The number of factors.
...	Arguments passed to rstan::sampling (e.g. iter, chains).

Value

A list.

train	Training data set.
gp_train	A vector of groups for each observation.
loadings	Aggregated loadings.
scores	Aggregated factor scores.
mu	Aggregated mean parameters.
probs	Aggregated probability parameters.
phi	Aggregated size parameters.
stan_mod	An object of S4 class stanfit.

gregorp90/RStan-package-test documentation built on May 26, 2019, 1:32 a.m.