def segmentation_loss()

in src/diarizers/models/model.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

    def segmentation_loss(
        self,
        permutated_prediction: torch.Tensor,
        target: torch.Tensor,
        weight: Optional[torch.Tensor] = None,