Skip to content

Commit 78a38de

Browse files
committed
varry fmax threshold as per paper
1 parent 58ae92d commit 78a38de

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

chebai/result/evaluate_predictions.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,8 @@ def calculate_fmax(
8080
Returns:
8181
Tuple[float, float]: The maximum F1 score and the corresponding threshold.
8282
"""
83-
thresholds = np.linspace(0, 1, 100)
83+
# https://github.com/bio-ontology-research-group/deepgo2/blob/main/deepgo/metrics.py#L51-L52
84+
thresholds = np.linspace(0, 1, 101)
8485
fmax = 0.0
8586
best_threshold = 0.0
8687

0 commit comments

Comments
 (0)