add test case for include_background=False

Signed-off-by: ytl0623 <[email protected]>

Author: ytl0623

tests/losses/test_unified_focal_loss.py

@@ -53,6 +53,18 @@
         },
         0.0,
     ],
+    # Case 2: Multi-class with background excluded
+    [
+        {
+            "use_softmax": True,
+            "include_background": False,
+        },
+        {
+            "y_pred": torch.tensor([[[[logit_pos, logit_neg]], [[logit_neg, logit_pos]], [[logit_neg, logit_neg]]]]),
+            "y_true": torch.tensor([[[[1.0, 0.0]], [[0.0, 1.0]], [[0.0, 0.0]]]]),
+        },
+        0.0,
+    ],
 ]