GOOD.networks.models.Classifiers

Applies a linear transformation to complete classification from representations.

Classes

Classifier(config)

Applies a linear transformation to complete classification from representations.

class GOOD.networks.models.Classifiers.Classifier(config: Union[CommonArgs, Munch])

Bases: Module

Applies a linear transformation to complete classification from representations.

Args:

config (Union[CommonArgs, Munch]): munchified dictionary of args (config.model.dim_hidden, config.dataset.num_classes)

forward(feat: Tensor) → Tensor

Applies a linear transformation to feature representations.

Parameters

feat (Tensor) – feature representations

Returns (Tensor):

label predictions