scglue.models.scclue.Discriminator

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

Bases: torch.nn.modules.container.Sequential, scglue.models.scclue.BatchedTransferrable

Methods

forward

Defines the computation performed at every call.

Attributes

batched_weights

List of batch-specific weights