sccross.models.layers.Classifier

class sccross.models.layers.Classifier(in_features, out_features, bias=True, device=None, dtype=None)

Bases: torch.nn.modules.linear.Linear

Linear label classifier

Parameters

• in_features (int) – Input dimensionality

• out_features (int) – Output dimensionality

Methods

Attributes