scglue.models.scclue.BernoulliDataDecoder

class scglue.models.scclue.BernoulliDataDecoder(in_features, out_features, batches=None, h_depth=2, h_dim=256, dropout=0.2)[source]

Bases: DataDecoder

Methods

compute_dist

rtype:

torch.distributions.bernoulli.Bernoulli

Attributes

batched_weights

List of batch-specific weights