scvi.module.base.BaseModuleClass

class scvi.module.base.BaseModuleClass[source]

Bases: torch.nn.modules.module.Module

Abstract class for scvi-tools modules.

Methods

forward(tensors[, …])

Forward pass through the network.

generative(*args, **kwargs)

Run the generative model.

inference(*args, **kwargs)

Run the inference (recognition) model.

loss(*args, **kwargs)

Compute the loss for a minibatch of data.

sample(*args, **kwargs)

Generate samples from the learned model.