From 339316aa7e892483e2a773117389e833b04eed94 Mon Sep 17 00:00:00 2001 From: Alexander Engelsberger Date: Tue, 17 May 2022 11:56:43 +0200 Subject: [PATCH] fix: use epsilon in cbc competition --- prototorch/core/competitions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/prototorch/core/competitions.py b/prototorch/core/competitions.py index 3cead31..f74eb28 100644 --- a/prototorch/core/competitions.py +++ b/prototorch/core/competitions.py @@ -38,7 +38,7 @@ def cbcc(detections: torch.Tensor, reasonings: torch.Tensor): pk = A nk = (1 - A) * B numerator = (detections @ (pk - nk).T) + nk.sum(1) - probs = numerator / (pk + nk).sum(1) + probs = numerator / ((pk + nk).sum(1) + 1e-8) return probs