scglue.models.sc.NBMixtureDataDecoder

class scglue.models.sc.NBMixtureDataDecoder(out_features, n_batches=1)[source]

Bases: DataDecoder

The Mixture of negative binomial data decoder

Parameters:

  • out_features (int) – Output dimensionality

  • n_batches (int) – Number of batches

Methods

forward

Decode data from sample and feature latent

Attributes