crowdkit.learning.conal.CoNAL.forward
| Source code
forward( self, embeddings: Tensor, logits: Tensor, workers: Tensor)
Forward pass of the CoNAL module.
Parameters | Type | Description |
---|---|---|
embeddings | Tensor | Tensor of shape (batch_size, embedding_size) |
logits | Tensor | Tensor of shape (batch_size, num_classes) |
workers | Tensor | Tensor of shape (batch_size,) containing the worker IDs. |
Returns:
Tensor of shape (batch_size, 1) containing the predicted output probabilities.
Return type:
Tensor
Last updated: March 31, 2023