Skip to content

Commit 01b772e

Browse files
sankovalevqubvel
authored andcommitted
Correction of an error in the function description (#35)
1 parent 02ef0e6 commit 01b772e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

segmentation_models_pytorch/utils/functions.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,11 @@ def f_score(pr, gt, beta=1, eps=1e-7, threshold=None, activation='sigmoid'):
4242
Args:
4343
pr (torch.Tensor): A list of predicted elements
4444
gt (torch.Tensor): A list of elements that are to be predicted
45+
beta (float): positive constant
4546
eps (float): epsilon to avoid zero division
4647
threshold: threshold for outputs binarization
4748
Returns:
48-
float: IoU (Jaccard) score
49+
float: F score
4950
"""
5051

5152
if activation is None or activation == "none":

0 commit comments

Comments
 (0)