Skip to content

Implements multi-label support for BCEWithLogitsLoss and MOONBCEWithLogitsLoss

André Anjos requested to merge issue-68-multi-label into main

This MR implements support for multi-label tasks (tasks in which multiple outputs of a classifier or segmenter may be positive concurrently).

Two losses are currently available to support binary, multi-class and multi-label tasks:

Tests have been added to the test suite.

Closes #68 (closed), biosignal/software/paper/euvip24-refine-cad-tb#2

Edited by André Anjos

Merge request reports