RumourDetectionTwitterModelOutput

class RumourDetectionTwitterModelOutput(loss: Optional[torch.Tensor] = None, logits: Optional[torch.Tensor] = None)

Base class for outputs of Rumour Detection models

Parameters

• loss (torch.Tensor of shape (1,), optional, returned when labels is provided) – Classification loss, typically cross entropy. Loss function used is dependent on what is specified in RumourDetectionTwitterConfig.

• logits (torch.Tensor of shape (batch_size, num_classes)) – Raw logits for each class. num_classes = 4 by default.