crowdkit.learning.conal.CoNAL.simple_common_module
| Source code
simple_common_module( self, input: Tensor, workers: Tensor)
Common noise adoptation module.
Parameters | Type | Description |
---|---|---|
input | Tensor | Tensor of shape (batch_size, embedding_size) |
workers | Tensor | Tensor of shape (batch_size,) containing the worker IDs. |
Returns:
Tensor of shape (batch_size, 1) containing the common noise rate.
Return type:
Tensor
Last updated: March 31, 2023