Skip to content

Commit 4f8725c

Browse files
committed
updated F1 test function
1 parent c5604f6 commit 4f8725c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_metrics.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def test_f1score():
2626

2727
target = torch.tensor([0, 1, 0, 2])
2828

29-
f1_metric.update(preds, target)
29+
f1_metric(preds, target)
3030
assert f1_metric.tp.sum().item() > 0, "Expected some true positives."
3131
assert f1_metric.fp.sum().item() > 0, "Expected some false positives."
3232
assert f1_metric.fn.sum().item() > 0, "Expected some false negatives."

0 commit comments

Comments
 (0)