julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: use epsilon in cbc competition

Browse Source
This commit is contained in:
Alexander Engelsberger 2022-05-17 11:56:43 +02:00
parent 2a85c94b55
commit 339316aa7e
No known key found for this signature in database
GPG Key ID: 72E54A9DAE51EB96
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/core/competitions.py
View File

@ -38,7 +38,7 @@ def cbcc(detections: torch.Tensor, reasonings: torch.Tensor):
pk = A pk = A
nk = (1 - A) * B nk = (1 - A) * B
numerator = (detections @ (pk - nk).T) + nk.sum(1) numerator = (detections @ (pk - nk).T) + nk.sum(1)
probs = numerator / (pk + nk).sum(1) probs = numerator / ((pk + nk).sum(1) + 1e-8)
return probs return probs