UFDClassifierModel

class UFDClassifierModel(config)

The UFD Classifier Model used for supervised training for source domain data.

This method inherits from UFDClassifierPreTrainedModel for weights initalization and utility functions from transformer PreTrainedModel class.

Parameters

config (UFDClassifierConfig) – Model configuration class with all parameters required for the model. Initializing with a config file does not load the weights associated with the model, only the configuration. Use the from_pretrained method to load the model weights.

forward(input_tensor: torch.Tensor) → torch.Tensor

Parameters

• input_tensor (torch.Tensor) – features from UFDCombineFeaturesMapModel

• labels (Optional[torch.Tensor], optional) – labels for input feature. Defaults to None.