Decoder¶
-
class
scvi.models.modules.
Decoder
(n_input, n_output, n_cat_list=None, n_layers=1, n_hidden=128)[source]¶ Bases:
torch.nn.modules.module.Module
Decodes data from latent space to data space
n_input
dimensions ton_output
dimensions using a fully-connected neural network ofn_hidden
layers. Output is the mean and variance of a multivariate Gaussian- Parameters
n_input (
int
int
) – The dimensionality of the input (latent space)n_output (
int
int
) – The dimensionality of the output (data space)n_cat_list (
Iterable
[int
],None
Optional
[Iterable
[int
]]) – A list containing the number of categories for each category of interest. Each category will be included using a one-hot encodingn_layers (
int
int
) – The number of fully-connected hidden layersdropout_rate – Dropout rate to apply to each of the hidden layers
- Returns
Methods Summary
forward
(x, *cat_list)The forward computation for a single sample.
Methods Documentation
-
forward
(x, *cat_list)[source]¶ The forward computation for a single sample.
Decodes the data from the latent space using the decoder network
Returns tensors for the mean and variance of a multivariate distribution
- Parameters
- Returns
2-tuple of
torch.Tensor
Mean and variance tensors of shape(n_output,)