scvi.module.AmortizedLDAPyroModule.get_elbo

AmortizedLDAPyroModule.get_elbo(x, library, n_obs)[source]

Computes ELBO.

Parameters
x : TensorTensor

Counts tensor.

library : TensorTensor

Library sizes for each cell.

n_obs : intint

Size of full batch. If n_obs < x.shape[0], ELBO is scaled by (n_obs / x.shape[0]).

Return type

floatfloat

Returns

The positive ELBO.