sccross.models.layers.NBDataEncoder

class sccross.models.layers.NBDataEncoder(in_features, out_features, h_depth=2, h_dim=256, dropout=0.2)[source]

Bases: torch.nn.modules.module.Module

Data encoder for negative binomial data

Parameters

  • in_features (int) – Input dimensionality

  • out_features (int) – Output dimensionality

  • h_depth (int) – Hidden layer depth

  • h_dim (int) – Hidden layer dimensionality

  • dropout (float) – Dropout rate

Methods

compute_l

rtype

Tensor

forward

Encode data to sample latent distribution

normalize

rtype

Tensor

Attributes

TOTAL_COUNT