scglue.models.scclue.DataDecoder

class scglue.models.scclue.DataDecoder(in_features, out_features, batches=None, h_depth=2, h_dim=256, dropout=0.2)

Bases: BatchedTransferrable

Methods

compute_dist	rtype: torch.distributions.distribution.Distribution
forward	Defines the computation performed at every call.

Attributes

batched_weights

List of batch-specific weights